Change agent hardcoded names to variables

kyma-project · Dec 11, 2024 · 20da8bc · 20da8bc
1 parent d2c4843
commit 20da8bc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/src/agents/supervisor/agent.py b/src/agents/supervisor/agent.py
@@ -11,7 +11,10 @@
 from langgraph.graph.graph import CompiledGraph
 
 from agents.common.constants import (
+    COMMON,
     FINALIZER,
+    K8S_AGENT,
+    KYMA_AGENT,
     MESSAGES,
     NEXT,
     PLANNER,
@@ -127,10 +130,7 @@ def _create_planner_chain(self, model: IModel) -> RunnableSequence:
                 ("system", PLANNER_PROMPT),
                 MessagesPlaceholder(variable_name="messages"),
             ]
-        ).partial(
-            members=self._get_members_str(),
-            output_format=self.plan_parser.get_format_instructions(),
-        )
+        ).partial(kymaAgent=KYMA_AGENT, kubernetesAgent=K8S_AGENT, commonAgent=COMMON)
         return self.planner_prompt | model.llm.with_structured_output(Plan)  # type: ignore
 
     async def _invoke_planner(self, state: SupervisorState) -> Plan:

diff --git a/src/agents/supervisor/prompts.py b/src/agents/supervisor/prompts.py
@@ -38,9 +38,9 @@
 - Focus solely on the key points raised in the query.
 
 Agent Classification:
-- "KymaAgent": Manages Kyma specific topics
-- "KubernetesAgent": Handles Kubernetes related queries
-- "Common": Covers all other general queries
+- "{kymaAgent}": Manages Kyma specific topics
+- "{kubernetesAgent}": Handles Kubernetes related queries
+- "{commonAgent}": Covers all other general queries
 
 
 Kyma terminologies: Kyma, Kubernetes, Serverless, Service Mesh, API Gateway, API Rule, Istio, Service Catalog, Application Connector, Eventing, Telemetry, Tracing, Logging, Kyma Runtime, module, Service Management.