Skip to content

Commit

Permalink
del print statements
Browse files Browse the repository at this point in the history
  • Loading branch information
sreyakumar committed Oct 25, 2024
1 parent c90ddbe commit 82f1771
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions src/metadata_chatbot/agents/workflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}

Expand Down Expand Up @@ -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'])
# inputs = {"query" : query}
# answer = app.invoke(inputs)
# print(answer['generation'])

0 comments on commit 82f1771

Please sign in to comment.