Aya1610 commited on
Commit
81b4c4b
·
verified ·
1 Parent(s): cdb890f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -2
app.py CHANGED
@@ -26,8 +26,22 @@ class BasicAgent:
26
  print(f"Agent received question (first 50 chars): {question[:50]}...")
27
  messages = [HumanMessage(content=question)]
28
  result = self.graph.invoke({"messages": messages})
29
- answer = result['messages'][-1].content
30
- return answer # kein [14:] mehr nötig!
 
 
 
 
 
 
 
 
 
 
 
 
 
 
31
 
32
 
33
 
 
26
  print(f"Agent received question (first 50 chars): {question[:50]}...")
27
  messages = [HumanMessage(content=question)]
28
  result = self.graph.invoke({"messages": messages})
29
+ print(f"Graph result: {result}")
30
+ if not isinstance(result, dict):
31
+ return "Graph returned an unexpected result format."
32
+
33
+ if 'messages' in result:
34
+ answer = result['messages'][-1].content
35
+ return answer
36
+ else:
37
+ return f"Graph returned: {result} (missing 'messages')"
38
+ # answer = result['messages'][-1].content
39
+ # return answer # kein [14:] mehr nötig!
40
+ # result = self.graph.invoke({"messages": messages})
41
+
42
+
43
+
44
+
45
 
46
 
47