From f64a33043232eb4aad93cd3a1668cfa7e1c4d032 Mon Sep 17 00:00:00 2001 From: sreyakumar <121137643+sreyakumar@users.noreply.github.com> Date: Thu, 21 Nov 2024 09:50:32 -0800 Subject: [PATCH] updated async_workflow --- src/metadata_chatbot/agents/async_workflow.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/metadata_chatbot/agents/async_workflow.py b/src/metadata_chatbot/agents/async_workflow.py index 06bf67e..6e24004 100644 --- a/src/metadata_chatbot/agents/async_workflow.py +++ b/src/metadata_chatbot/agents/async_workflow.py @@ -1,14 +1,16 @@ -import asyncio, json +import asyncio from typing import List, Optional from typing_extensions import TypedDict + from langchain_core.documents import Document from langgraph.graph import END, StateGraph, START -from docdb_retriever import DocDBRetriever -from metadata_chatbot.agents.docdb_retriever import DocDBRetriever -from react_agent import react_agent from langchain_core.messages.ai import AIMessage + +from metadata_chatbot.agents.docdb_retriever import DocDBRetriever +from metadata_chatbot.agents.react_agent import react_agent from metadata_chatbot.agents.agentic_graph import datasource_router, filter_generation_chain, doc_grader, rag_chain, db_rag_chain + class GraphState(TypedDict): """ Represents the state of our graph.