Merge pull request m-cmp#154 from kyuengmanKim/develop/log-analysis

Develop/log analysis

Author: kyuengmanKim

python/insight/app/core/mcp/mcp_context.py

@@ -47,17 +47,24 @@ async def get_agent(self, provider: str, model_name: str, provider_credential: s
         self.llm_client.setup(model=model_name)
         self.agent = self.llm_client.bind_tools(self.tools, self.memory)
 
+    async def _build_prompt(self, session_id: str, messages: str):
+        history = await self.get_chat_history(session_id)
+        msg_count = 0
+        if history:
+            channel_values = history.get('channel_values', {})
+            msg_count = len(channel_values.get('messages', []))
 
-    @staticmethod
-    def _build_prompt(messages):
         current_time = datetime.now(UTC).strftime("%Y-%m-%dT%H:%M:%SZ")
 
-        system_prompt = f"""
-                You are an assistant specialized in log retrieval and log exploration.
-                To respond to user questions, select the appropriate tools and provide thoughtful and sincere answers. 
-                If the question is in English, respond in English; if it's in Korean, respond in Korean.
-                The current time is {current_time}.
-                """
+        # TODO 기능, 상황별 Prompt 관리 기능 추가
+        system_prompt_config = self.config.get_system_prompt_config()
+
+        if msg_count == 0:
+            system_prompt_first = system_prompt_config.get("system_prompt_first")
+            system_prompt = system_prompt_first.format(current_time=current_time)
+        else:
+            system_prompt_default = system_prompt_config.get("system_prompt_default")
+            system_prompt = system_prompt_default.format(current_time=current_time)
 
         return [
             {"role": "system", "content": system_prompt},
@@ -66,12 +73,10 @@ def _build_prompt(messages):
 
     async def aquery(self, session_id, messages):
         config = self.create_config(session_id)
-        prompt = self._build_prompt(messages)
+        prompt = await self._build_prompt(session_id, messages)
         response = await self.agent.ainvoke({'messages': prompt}, config=config)
-
         return response
 
-
     @staticmethod
     def create_config(session_id):
         return {'configurable': {'thread_id': f'{session_id}'}}

python/insight/config/ConfigManager.py

@@ -86,4 +86,11 @@ def get_mcp_config(self):
         mcp = self.config.get('log_analysis', {}).get('mcp', {})
         return {
             "mcp_grafana_url": mcp.get('mcp_grafana_url', '')
+        }
+
+    def get_system_prompt_config(self):
+        log_analysis = self.config.get('log_analysis', {})
+        return {
+            "system_prompt_first": log_analysis.get('system_prompt_first', ''),
+            "system_prompt_default": log_analysis.get('system_prompt_default', '')
         }

python/insight/config/config.yaml

@@ -116,4 +116,13 @@ log_analysis:
         - gpt-4o-mini
         - gpt-3.5-turbo
   mcp:
-    mcp_grafana_url: http://mc-observability-mcp-grafana:8000/sse
+    mcp_grafana_url: http://mc-observability-mcp-grafana:8000/sse
+  system_prompt_first: |
+    You are an assistant specialized in log retrieval and log exploration.
+    To respond to user questions, select the appropriate tools and provide thoughtful and sincere answers. 
+    If the question is in English, respond in English; if it's in Korean, respond in Korean.
+    The current time is {current_time}.
+  system_prompt_default: |
+    You are an assistant specialized in log retrieval and log exploration.
+    respond in Korean.
+    Continue the conversation. The current time is {current_time}.