Fix

langchain-ai · May 6, 2024 · 4aef884 · 4aef884
1 parent ad4ce87
commit 4aef884
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/backend/app/graphs/new_rag.py b/backend/app/graphs/new_rag.py
@@ -104,7 +104,11 @@ def _get_messages(messages, system_message=DEFAULT_SYSTEM_MESSAGE):
 
 @chain
 async def get_search_query(messages: Sequence[BaseMessage], config):
-    llm = config["configurable"].get("llm_type", LLMType.GPT_35_TURBO)
+    llm = get_llm(
+        config["configurable"].get(
+            "agent==chat_retrieval/llm_type", LLMType.GPT_35_TURBO
+        )
+    )
     convo = []
     for m in messages:
         if isinstance(m, AIMessage):