updated demo

.gitignore

@@ -2,6 +2,7 @@
 chatbot.ipynb
 *.pkl
 *.png
+queries.txt
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

src/metadata_chatbot/agents/GAMER.py

@@ -111,7 +111,8 @@ def _llm_type(self) -> str:
         """Get the type of language model used by this chat model. Used for logging purposes only."""
         return "Claude 3 Sonnet"
 
-
+# llm = GAMER()
+# print(llm.invoke("Can you give me a timeline of events for subject 675387?"))
 
 # async def main():
 #     result = await llm.ainvoke("Can you give me a timeline of events for subject 675387?")

src/metadata_chatbot/agents/async_workflow.py

@@ -191,7 +191,7 @@ 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?"
+#     query = "Can you give me a timeline of events for subject 675387?"
 #     inputs = {"query": query}
 #     result = async_app.astream(inputs)
 
@@ -203,5 +203,5 @@ async def generate_async(state):
 #     if value:
 #         print(value['generation'])
 
-# # Run the async function
+# #Run the async function
 # asyncio.run(main())

src/metadata_chatbot/agents/workflow.py

@@ -186,7 +186,7 @@ def generate(state):
 
 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"
+# query = "Give me the names of 5 assets have injections and are smartspim?"
 
 # inputs = {"query" : query}
 # answer = app.invoke(inputs)

src/metadata_chatbot/bedrock_model/ref/session_schema.json

@@ -1,5 +1,4 @@
-{
-
+{       
    "additionalProperties": false,
    "description": "Description of a physiology and/or behavior session",
    "properties": {