Shamik commited on
Commit
b38c2ca
·
verified ·
1 Parent(s): bc057a1

feat: adding logger for manager agent prompt.

Browse files

fix: modifying the prefix prompt to return the outputs of the insurance agent.

src/insurance_assistants/agents.py CHANGED
@@ -134,4 +134,4 @@ manager_agent = CodeAgent(
134
  "health insurance documents using the `insurance_agent`, search wikipedia and the web for general information.",
135
  )
136
  manager_agent.system_prompt = manager_agent.system_prompt + PROMPT_PREFIX
137
- print(manager_agent.system_prompt)
 
134
  "health insurance documents using the `insurance_agent`, search wikipedia and the web for general information.",
135
  )
136
  manager_agent.system_prompt = manager_agent.system_prompt + PROMPT_PREFIX
137
+
src/insurance_assistants/consts.py CHANGED
@@ -26,7 +26,7 @@ PROMPT_PREFIX = """\n
26
  Apart from all the above instructions that we have given to you, FOLLOW the Additional Instructions below:
27
  ```
28
  For any health insurance related queries, always use the `insurance_agent` first and return the results.
29
- You are allowed to rephrase any query and detail it if required. When in doubt always ask the user a follow up question. No need to cross reference the `insurance_agent` results with web search. Return the results from the `insurance_agent` directly.
30
  Don't assume anything.
31
  ```
32
 
 
26
  Apart from all the above instructions that we have given to you, FOLLOW the Additional Instructions below:
27
  ```
28
  For any health insurance related queries, always use the `insurance_agent` first and return the results.
29
+ Don't use any other agent but the `insurance_agent`. Return the results from the `insurance_agent` directly.
30
  Don't assume anything.
31
  ```
32
 
src/insurance_assistants/ui.py CHANGED
@@ -91,6 +91,7 @@ class UI:
91
  if "agent" not in session_state:
92
  # session_state["agent"] = CodeAgent(tools=[], model=InfenceClientModel())
93
  session_state["agent"] = manager_agent
 
94
  # Adding monitoring
95
  try:
96
  # log the existence of agent memory
 
91
  if "agent" not in session_state:
92
  # session_state["agent"] = CodeAgent(tools=[], model=InfenceClientModel())
93
  session_state["agent"] = manager_agent
94
+ logger.info(f"Manager agent prompt: {manager_agent.system_prompt}")
95
  # Adding monitoring
96
  try:
97
  # log the existence of agent memory