Spaces:
Running
Running
fix: always adding the prefix prompt for all the agent sessions in gradio.
Browse files
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 |
-
|
|
|
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
|