IamRulo commited on
Commit
8a7741b
·
verified ·
1 Parent(s): 2e8edbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -21,22 +21,18 @@ class BasicAgent:
21
  """A LangGraph agent"""
22
  def __init__(self):
23
  print("BasicAgent initialized.")
24
- #rulo
25
  self.graph = build_graph("huggingface")
26
 
27
  def __call__(self, question: str) -> str:
28
  print(f"Agent received question (first 50 chars): {question[:50]}...")
29
- #rulo
30
  # Wrap the question in a HumanMessage from langchain_core
31
- #messages = [HumanMessage(content=question)]
32
- messages = [sys_msg + HumanMessage(content=question)]
33
  messages = self.graph.invoke({"messages": messages})
34
  answer = messages['messages'][-1].content
35
 
36
  print(f"answer: {answer}") #rulo
37
  return answer[14:]
38
- #rulo
39
-
40
 
41
  def run_and_submit_all( profile: gr.OAuthProfile | None):
42
  """
 
21
  """A LangGraph agent"""
22
  def __init__(self):
23
  print("BasicAgent initialized.")
 
24
  self.graph = build_graph("huggingface")
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
 
33
  print(f"answer: {answer}") #rulo
34
  return answer[14:]
35
+
 
36
 
37
  def run_and_submit_all( profile: gr.OAuthProfile | None):
38
  """