From 82f17714dd787fa78fc350b70d6aefbf15d6e2d2 Mon Sep 17 00:00:00 2001
From: sreyakumar <121137643+sreyakumar@users.noreply.github.com>
Date: Fri, 25 Oct 2024 13:52:21 -0700
Subject: [PATCH] del print statements

---
 src/metadata_chatbot/agents/workflow.py | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/metadata_chatbot/agents/workflow.py b/src/metadata_chatbot/agents/workflow.py
index 53b2579..6822b8d 100644
--- a/src/metadata_chatbot/agents/workflow.py
+++ b/src/metadata_chatbot/agents/workflow.py
@@ -68,8 +68,6 @@ def generate_for_whole_db(state):
     document_dict['mongodb_query'] = retrieved_dict['intermediate_steps'][0][0].tool_input['agg_pipeline']
     document_dict['retrieved_output'] = retrieved_dict['intermediate_steps'][0][1]
 
-    print(document_dict)
-
     documents = json.dumps(document_dict)
     return {"query": query, "documents": documents}
 
@@ -195,8 +193,8 @@ def generate(state):
 
 app = workflow.compile()
 
-query = "What is the mongodb query to find all the assets using mouse 675387"
+# query = "What is the mongodb query to find all the assets using mouse 675387"
 
-inputs = {"query" : query}
-answer = app.invoke(inputs)
-print(answer['generation'])
\ No newline at end of file
+# inputs = {"query" : query}
+# answer = app.invoke(inputs)
+# print(answer['generation'])
\ No newline at end of file