Shamik commited on
Commit
2419faa
·
verified ·
1 Parent(s): 5f073d9

fix: always adding the prefix prompt for all the agent sessions in gradio.

Browse files
Files changed (1) hide show
  1. src/insurance_assistants/ui.py +3 -2
src/insurance_assistants/ui.py CHANGED
@@ -13,7 +13,7 @@ from smolagents.models import ChatMessageStreamDelta
13
 
14
  # from smolagents import CodeAgent, InferenceClientModel
15
  from src.insurance_assistants.agents import manager_agent
16
- from src.insurance_assistants.consts import PRIMARY_HEADING, PROJECT_ROOT_DIR
17
 
18
  load_dotenv(override=True)
19
 
@@ -91,7 +91,8 @@ class UI:
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
 
13
 
14
  # from smolagents import CodeAgent, InferenceClientModel
15
  from src.insurance_assistants.agents import manager_agent
16
+ from src.insurance_assistants.consts import PRIMARY_HEADING, PROJECT_ROOT_DIR, PROMPT_PREFIX
17
 
18
  load_dotenv(override=True)
19
 
 
91
  if "agent" not in session_state:
92
  # session_state["agent"] = CodeAgent(tools=[], model=InfenceClientModel())
93
  session_state["agent"] = manager_agent
94
+ session_state["agent"].system_prompt = session_state["agent"].system_prompt + PROMPT_PREFIX
95
+
96
  # Adding monitoring
97
  try:
98
  # log the existence of agent memory