anandshende-videocx commited on
Commit
f1c760a
·
verified ·
1 Parent(s): b0809d3

Update my_agent.py

Browse files
Files changed (1) hide show
  1. my_agent.py +2 -2
my_agent.py CHANGED
@@ -126,7 +126,7 @@ class BasicAgent:
126
  # )
127
  response = self.llm_qa.invoke(
128
  parsed_messages,
129
- {"callbacks": [langfuse_handler]},
130
  )
131
  print(f"LLAMA 2 -> QA Agent raw response: {response}")
132
  return response.model_dump()
@@ -146,7 +146,7 @@ class BasicAgent:
146
  ],
147
  "question": question,
148
  },
149
- {"callbacks": [langfuse_handler]},
150
  )
151
  print(f"Agent raw response: {response}")
152
  return response["messages"][-1].content
 
126
  # )
127
  response = self.llm_qa.invoke(
128
  parsed_messages,
129
+ # {"callbacks": [langfuse_handler]},
130
  )
131
  print(f"LLAMA 2 -> QA Agent raw response: {response}")
132
  return response.model_dump()
 
146
  ],
147
  "question": question,
148
  },
149
+ # {"callbacks": [langfuse_handler]},
150
  )
151
  print(f"Agent raw response: {response}")
152
  return response["messages"][-1].content