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)
|