Datawithsarah commited on
Commit
0c02b02
·
1 Parent(s): c81aa98

formatter function

Browse files
Files changed (1) hide show
  1. agent.py +5 -5
agent.py CHANGED
@@ -116,12 +116,12 @@ def build_graph(provider: str = "claude"):
116
  return {"messages": state["messages"] + [response]}
117
 
118
  def formatter(state: MessagesState):
119
- last = state["messages"][-1].content.strip()
120
 
121
- cleaned = re.sub(r"<.*?>", "", last)
122
- cleaned = re.sub(r"(Final\s*Answer:|Answer:)", "", cleaned, flags=re.IGNORECASE)
123
- cleaned = cleaned.strip().split("\n")[0].strip()
124
- return {"messages": [AIMessage(content=cleaned)]}
125
 
126
  builder = StateGraph(MessagesState)
127
  builder.add_node("retriever", retriever)
 
116
  return {"messages": state["messages"] + [response]}
117
 
118
  def formatter(state: MessagesState):
119
+ last = state["messages"][-1].content.strip()
120
 
121
+ cleaned = re.sub(r"<.*?>", "", last)
122
+ cleaned = re.sub(r"(Final\s*Answer:|Answer:)", "", cleaned, flags=re.IGNORECASE)
123
+ cleaned = cleaned.strip().split("\n")[0].strip()
124
+ return {"messages": [AIMessage(content=cleaned)]}
125
 
126
  builder = StateGraph(MessagesState)
127
  builder.add_node("retriever", retriever)