ZeroTimo commited on
Commit
47d9afe
·
verified ·
1 Parent(s): 241ab03

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,11 +25,11 @@ class BasicAgent:
25
 
26
  def __call__(self, question: str) -> str:
27
  print(f"Agent received question (first 50 chars): {question[:50]}...")
28
- # Wrap the question in a HumanMessage from langchain_core
29
  messages = [HumanMessage(content=question)]
30
- messages = self.graph.invoke({"messages": messages})
31
- answer = messages['messages'][-1].content
32
- return answer[14:]
 
33
 
34
 
35
  def run_and_submit_all( profile: gr.OAuthProfile | None):
 
25
 
26
  def __call__(self, question: str) -> str:
27
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
28
  messages = [HumanMessage(content=question)]
29
+ result = self.graph.invoke({"messages": messages})
30
+ answer = result['messages'][-1].content
31
+ return answer # kein [14:] mehr nötig!
32
+
33
 
34
 
35
  def run_and_submit_all( profile: gr.OAuthProfile | None):