Commit
·
0c02b02
1
Parent(s):
c81aa98
formatter function
Browse files
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 |
-
|
120 |
|
121 |
-
|
122 |
-
|
123 |
-
|
124 |
-
|
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)
|