diff --git a/.gitignore b/.gitignore index b3fcf851..b13025e8 100644 --- a/.gitignore +++ b/.gitignore @@ -155,3 +155,4 @@ node_modules/ notebooks .langgraph-data/ +.langgraph_api/ \ No newline at end of file diff --git a/backend/retrieval_graph/graph.py b/backend/retrieval_graph/graph.py index ffa787fb..3ace8208 100644 --- a/backend/retrieval_graph/graph.py +++ b/backend/retrieval_graph/graph.py @@ -144,7 +144,9 @@ class Plan(TypedDict): messages = [ {"role": "system", "content": configuration.research_plan_system_prompt} ] + state.messages - response = cast(Plan, await model.ainvoke(messages)) + response = cast( + Plan, await model.ainvoke(messages, {"tags": ["langsmith:nostream"]}) + ) return { "steps": response["steps"], "documents": "delete", diff --git a/backend/retrieval_graph/researcher_graph/graph.py b/backend/retrieval_graph/researcher_graph/graph.py index 6ade95b1..28ecba5f 100644 --- a/backend/retrieval_graph/researcher_graph/graph.py +++ b/backend/retrieval_graph/researcher_graph/graph.py @@ -42,7 +42,9 @@ class Response(TypedDict): {"role": "system", "content": configuration.generate_queries_system_prompt}, {"role": "human", "content": state.question}, ] - response = cast(Response, await model.ainvoke(messages)) + response = cast( + Response, await model.ainvoke(messages, {"tags": ["langsmith:nostream"]}) + ) return {"queries": response["queries"]}