sampsong commited on
Commit
14b3291
·
1 Parent(s): 056177a

add prompt

Browse files
Files changed (2) hide show
  1. Agents/agent.py +3 -3
  2. 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(str(msg))]}
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
- messages = self.graph.invoke({"messages": messages})
 
 
 
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]