Update my_agent.py
Browse files- 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
|