xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-02 20:00:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update langchain

Browse Source
This commit is contained in:
Brian Pham
2023-12-12 23:28:58 +08:00
parent 0c66554d50
commit a627dcd64f
2 changed files with 13 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/algo/utils.py
View File

@ -11,7 +11,7 @@ import yaml
from pr_agent.algo.ai_handlers.litellm_ai_handler import LiteLLMAiHandler
from pr_agent.algo.ai_handlers.base_ai_handler import BaseAiHandler
from pr_agent.algo.ai_handlers.openai_ai_handler import OpenAIHandler
from pr_agent.algo.ai_handlers.langchain_ai_handler import LangChainAIHandler
from pr_agent.algo.ai_handlers.langchain_ai_handler import LangChainOpenAIHandler
from starlette_context import context
from pr_agent.config_loader import get_settings, global_settings
from pr_agent.log import get_logger
@ -310,4 +310,4 @@ def try_fix_yaml(review_text: str) -> dict:
return data
def get_ai_handler() -> BaseAiHandler:
return LangChainAIHandler()
return LangChainOpenAIHandler()