Spaces:
Sleeping
Sleeping
add prompt
Browse files- Agents/agent.py +3 -3
- app.py +4 -1
Agents/agent.py
CHANGED
|
@@ -240,12 +240,12 @@ def build_graph(provider: str="groq"):
|
|
| 240 |
|
| 241 |
def assistant(state: MessagesState):
|
| 242 |
#msg
|
| 243 |
-
msg = {"messages": [sysMsg] + state["messages"]}
|
| 244 |
|
| 245 |
#print(f"invoke llmwithtools with state message = {msg}")
|
| 246 |
#msgbind = llmWithTools.invoke(state["messages"])
|
| 247 |
-
print(f"msg: {msg}")
|
| 248 |
-
return {"messages": [llmWithTools.invoke(
|
| 249 |
'''
|
| 250 |
def retriever(state: MessagesState):
|
| 251 |
similiarQuestion = vector_store.similarity_search(state["message"][0].content)
|
|
|
|
| 240 |
|
| 241 |
def assistant(state: MessagesState):
|
| 242 |
#msg
|
| 243 |
+
#msg = {"messages": [sysMsg] + state["messages"]}
|
| 244 |
|
| 245 |
#print(f"invoke llmwithtools with state message = {msg}")
|
| 246 |
#msgbind = llmWithTools.invoke(state["messages"])
|
| 247 |
+
#print(f"msg: {msg}")
|
| 248 |
+
return {"messages": [llmWithTools.invoke(state["message"])]}
|
| 249 |
'''
|
| 250 |
def retriever(state: MessagesState):
|
| 251 |
similiarQuestion = vector_store.similarity_search(state["message"][0].content)
|
app.py
CHANGED
|
@@ -60,7 +60,10 @@ class BasicAgent:
|
|
| 60 |
|
| 61 |
if(not langFuseOn):
|
| 62 |
print("no langfuse")
|
| 63 |
-
|
|
|
|
|
|
|
|
|
|
| 64 |
else:
|
| 65 |
print("active langfuse")
|
| 66 |
formattedMessage = [{"role":"user","content":msg.content} for msg in messages]
|
|
|
|
| 60 |
|
| 61 |
if(not langFuseOn):
|
| 62 |
print("no langfuse")
|
| 63 |
+
formattedMessage = [{"role":"user","content":msg.content} for msg in messages]
|
| 64 |
+
formattedMessage.append(SystemMessage(content="systemPrompt"))
|
| 65 |
+
print(f"formattedMessage : {formattedMessage}")
|
| 66 |
+
messages = self.graph.invoke({"messages": formattedMessage})
|
| 67 |
else:
|
| 68 |
print("active langfuse")
|
| 69 |
formattedMessage = [{"role":"user","content":msg.content} for msg in messages]
|