Merge pull request #2 from AllenNeuralDynamics/dev

added demo ipynb

demo.ipynb

@@ -2,21 +2,31 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
-   "metadata": {
-    "vscode": {
-     "languageId": "plaintext"
-    }
-   },
+   "execution_count": 5,
+   "metadata": {},
    "outputs": [],
    "source": [
-    "from metadata_chatbot.agents"
+    "from metadata_chatbot.agents.gamer import GAMER"
    ]
   }
  ],
  "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
   "language_info": {
-   "name": "python"
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.9.13"
   }
  },
  "nbformat": 4,

src/metadata_chatbot/agents/GAMER.py

@@ -8,8 +8,8 @@
 
 import logging, asyncio
 
-from async_workflow import async_app
-from workflow import app
+from metadata_chatbot.agents.async_workflow import async_app
+from metadata_chatbot.agents.workflow import app
 
 
 class GAMER(LLM):

src/metadata_chatbot/agents/async_workflow.py

@@ -9,8 +9,8 @@
 sys.path.append(os.path.abspath("C:/Users/sreya.kumar/Documents/GitHub/metadata-chatbot"))
 from metadata_chatbot.utils import ResourceManager
 
-from docdb_retriever import DocDBRetriever
-from agentic_graph import datasource_router, db_surveyor, query_grader, filter_generation_chain, doc_grader, rag_chain
+from metadata_chatbot.agents.docdb_retriever import DocDBRetriever
+from metadata_chatbot.agents.agentic_graph import datasource_router, db_surveyor, query_grader, filter_generation_chain, doc_grader, rag_chain
 
 logging.basicConfig(filename='async_workflow.log', level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', filemode="w")
 
@@ -190,18 +190,18 @@ async def generate_async(state):
 
 async_app = async_workflow.compile()
 
-async def main():
-    query = "What was the age of the subject when receiving injections in asset SmartSPIM_675388_2023-05-24_04-10-19_stitched_2023-05-28_18-07-46?"
-    inputs = {"query": query}
-    result = async_app.astream(inputs)
+# async def main():
+#     query = "What was the age of the subject when receiving injections in asset SmartSPIM_675388_2023-05-24_04-10-19_stitched_2023-05-28_18-07-46?"
+#     inputs = {"query": query}
+#     result = async_app.astream(inputs)
 
-    value = None
-    async for output in result:
-        for key, value in output.items():
-            logging.info(f"Currently on node '{key}':")
+#     value = None
+#     async for output in result:
+#         for key, value in output.items():
+#             logging.info(f"Currently on node '{key}':")
 
-    if value:
-        print(value['generation'])
+#     if value:
+#         print(value['generation'])
 
-# Run the async function
-asyncio.run(main())
+# # Run the async function
+# asyncio.run(main())

src/metadata_chatbot/agents/workflow.py

@@ -8,8 +8,8 @@
 sys.path.append(os.path.abspath("C:/Users/sreya.kumar/Documents/GitHub/metadata-chatbot"))
 from metadata_chatbot.utils import ResourceManager
 
-from docdb_retriever import DocDBRetriever
-from agentic_graph import datasource_router, db_surveyor, query_grader, filter_generation_chain, doc_grader, rag_chain
+from metadata_chatbot.agents.docdb_retriever import DocDBRetriever
+from metadata_chatbot.agents.agentic_graph import datasource_router, db_surveyor, query_grader, filter_generation_chain, doc_grader, rag_chain
 
 logging.basicConfig(filename='async_workflow.log', level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', filemode="w")
 
@@ -183,8 +183,8 @@ def generate(state):
 workflow.add_edge("document_grading","generate")
 workflow.add_edge("generate", END)
 
-memory = MemorySaver()
-app = workflow.compile(checkpointer=memory)
+
+app = workflow.compile()
 
 # query = "How old was the subject in SmartSPIM_675388_2023-05-24_04-10-19_stitched_2023-05-28_18-07-46"