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

langchain: move model and temperature to chat_completion

Browse Source
This commit is contained in:
Brian Pham
2023-12-12 23:07:46 +08:00
parent 506eafc0c5
commit 0c66554d50
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pr_agent/algo/ai_handlers/langchain_ai_handler.py
View File

@ -12,7 +12,7 @@ from pr_agent.config_loader import get_settings
from pr_agent.log import get_logger from pr_agent.log import get_logger
OPENAI_RETRIES = 5 OPENAI_RETRIES = 5
chat = ChatOpenAI(openai_api_key = get_settings().openai.key, model="gpt-4") chat = ChatOpenAI(openai_api_key = get_settings().openai.key)
class LangChainAIHandler(BaseAiHandler): class LangChainAIHandler(BaseAiHandler):
def __init__(self): def __init__(self):
@ -35,7 +35,7 @@ class LangChainAIHandler(BaseAiHandler):
messages=[SystemMessage(content=system), HumanMessage(content=user)] messages=[SystemMessage(content=system), HumanMessage(content=user)]
# get a chat completion from the formatted messages # get a chat completion from the formatted messages
resp = chat(messages) resp = chat(messages, model=model, temperature=temperature)
get_logger().info("AI response: ", resp.content) get_logger().info("AI response: ", resp.content)
finish_reason="completed" finish_reason="completed"
return resp.content, finish_reason return resp.content, finish_reason