From 0c66554d505a1a607e0cc14bc80e4c750bcad13c Mon Sep 17 00:00:00 2001 From: Brian Pham Date: Tue, 12 Dec 2023 23:07:46 +0800 Subject: [PATCH] langchain: move model and temperature to chat_completion --- pr_agent/algo/ai_handlers/langchain_ai_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pr_agent/algo/ai_handlers/langchain_ai_handler.py b/pr_agent/algo/ai_handlers/langchain_ai_handler.py index 406c6c40..87cf1072 100644 --- a/pr_agent/algo/ai_handlers/langchain_ai_handler.py +++ b/pr_agent/algo/ai_handlers/langchain_ai_handler.py @@ -12,7 +12,7 @@ from pr_agent.config_loader import get_settings from pr_agent.log import get_logger 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): def __init__(self): @@ -35,7 +35,7 @@ class LangChainAIHandler(BaseAiHandler): messages=[SystemMessage(content=system), HumanMessage(content=user)] # 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) finish_reason="completed" return resp.content, finish_reason