mikethor007 commited on
Commit
4f281de
·
verified ·
1 Parent(s): aaca609

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -99,14 +99,16 @@ class BasicAgent:
99
  def __call__(self, question: str) -> str:
100
  print(f"Agent received question (first 50 chars): {question[:50]}...")
101
 
102
- messages = [
103
- SystemMessage(content="You're a helpful assistant"),
104
- HumanMessage(
105
- content=question
106
- ),
107
- ]
108
-
109
- answer = chat.invoke(messages)
 
 
110
  print(f"Agent returning answer: {answer}")
111
  return answer
112
 
 
99
  def __call__(self, question: str) -> str:
100
  print(f"Agent received question (first 50 chars): {question[:50]}...")
101
 
102
+ messages = {
103
+ "messages": [
104
+ {
105
+ "role": "user",
106
+ "content": question
107
+ }
108
+ ]
109
+ }
110
+
111
+ answer = supervisor.invoke(messages)
112
  print(f"Agent returning answer: {answer}")
113
  return answer
114