sid385 commited on
Commit
ecf2e9a
·
verified ·
1 Parent(s): 0a72ca9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -57,7 +57,12 @@ class BasicAgent:
57
 
58
  def model_call(state: AgentState) -> AgentState:
59
  system_prompt = SystemMessage(
60
- content="You are an AI assistant. Use tools like math functions and web search (DuckDuckGo) to answer queries."
 
 
 
 
 
61
  )
62
  response = self.model.invoke([system_prompt] + state["messages"])
63
  return {"messages": [response]}
 
57
 
58
  def model_call(state: AgentState) -> AgentState:
59
  system_prompt = SystemMessage(
60
+ content="""
61
+ You are an AI assistant. Use tools like math functions and web search (DuckDuckGo) to answer queries.
62
+ You don't have to explain or give reasons just you have to answer the question
63
+ Just reply the question eg.
64
+ if asked about what is capital of France just answer paris nothing more
65
+ """
66
  )
67
  response = self.model.invoke([system_prompt] + state["messages"])
68
  return {"messages": [response]}