hdsp-jupyter-extension 2.0.10__py3-none-any.whl → 2.0.11__py3-none-any.whl

hdsp_agent_core/services/agent_service.py

@@ -179,11 +179,11 @@ class EmbeddedAgentService(IAgentService):
         rag_context = None
         try:
             from hdsp_agent_core.factory import get_service_factory
+
             rag_service = get_service_factory().get_rag_service()
             if rag_service.is_ready():
                 rag_context = await rag_service.get_context_for_query(
-                    query=request.request,
-                    detected_libraries=detected_libraries
+                    query=request.request, detected_libraries=detected_libraries
                 )
         except Exception as e:
             logger.warning(f"RAG context retrieval failed: {e}")
@@ -208,7 +208,9 @@ class EmbeddedAgentService(IAgentService):
         plan_data = self._parse_json_response(response)
 
         if not plan_data or "plan" not in plan_data:
-            raise ValueError(f"Failed to parse plan from LLM response: {response[:200]}")
+            raise ValueError(
+                f"Failed to parse plan from LLM response: {response[:200]}"
+            )
 
         # Sanitize and ensure goal
         plan_data = self._sanitize_tool_calls(plan_data)
@@ -291,6 +293,7 @@ class EmbeddedAgentService(IAgentService):
         # Use error classifier for deterministic classification
         try:
             from hdsp_agent_core.managers.error_classifier import get_error_classifier
+
             classifier = get_error_classifier()
         except ImportError:
             # Fallback if error_classifier not yet migrated
@@ -346,7 +349,9 @@ class EmbeddedAgentService(IAgentService):
             return {
                 "valid": result.is_valid,
                 "issues": [issue.to_dict() for issue in result.issues],
-                "dependencies": result.dependencies.to_dict() if result.dependencies else None,
+                "dependencies": result.dependencies.to_dict()
+                if result.dependencies
+                else None,
                 "hasErrors": result.has_errors,
                 "hasWarnings": result.has_warnings,
                 "summary": result.summary,
@@ -379,10 +384,7 @@ class ProxyAgentService(IAgentService):
         logger.info(f"ProxyAgentService initialized (server: {self._base_url})")
 
     async def _request(
-        self,
-        method: str,
-        path: str,
-        data: Optional[Dict] = None
+        self, method: str, path: str, data: Optional[Dict] = None
     ) -> Dict[str, Any]:
         """Make HTTP request to agent server"""
         url = f"{self._base_url}{path}"

hdsp_agent_core/services/chat_service.py

@@ -69,7 +69,9 @@ class EmbeddedChatService(IChatService):
         session = self._session_manager.get_or_create_session(conversation_id)
         return session.id
 
-    def _build_context(self, conversation_id: str, max_messages: int = 5) -> Optional[str]:
+    def _build_context(
+        self, conversation_id: str, max_messages: int = 5
+    ) -> Optional[str]:
         """Build conversation context from history"""
         return self._session_manager.build_context(conversation_id, max_messages)
 
@@ -88,7 +90,9 @@ class EmbeddedChatService(IChatService):
         config = self._build_llm_config(request.llmConfig)
 
         if not config or not config.get("provider"):
-            raise ValueError("LLM not configured. Please provide llmConfig with API keys.")
+            raise ValueError(
+                "LLM not configured. Please provide llmConfig with API keys."
+            )
 
         # Get or create conversation
         conversation_id = self._get_or_create_conversation(request.conversationId)
@@ -98,9 +102,7 @@ class EmbeddedChatService(IChatService):
 
         # Call LLM
         llm_service = LLMService(config)
-        response = await llm_service.generate_response(
-            request.message, context=context
-        )
+        response = await llm_service.generate_response(request.message, context=context)
 
         # Store messages
         self._store_messages(conversation_id, request.message, response)
@@ -124,7 +126,9 @@ class EmbeddedChatService(IChatService):
         config = self._build_llm_config(request.llmConfig)
 
         if not config or not config.get("provider"):
-            yield {"error": "LLM not configured. Please provide llmConfig with API keys."}
+            yield {
+                "error": "LLM not configured. Please provide llmConfig with API keys."
+            }
             return
 
         # Get or create conversation

hdsp_agent_core/services/rag_service.py

@@ -91,7 +91,7 @@ class EmbeddedRAGService(IRAGService):
         self,
         query: str,
         detected_libraries: Optional[List[str]] = None,
-        max_results: int = 5
+        max_results: int = 5,
     ) -> Optional[str]:
         """Get formatted context for a query"""
         logger.info(f"[Embedded] Get context for query: {query[:100]}...")
@@ -186,10 +186,7 @@ class ProxyRAGService(IRAGService):
         return self._ready
 
     async def _request(
-        self,
-        method: str,
-        path: str,
-        data: Optional[Dict] = None
+        self, method: str, path: str, data: Optional[Dict] = None
     ) -> Dict[str, Any]:
         """Make HTTP request to RAG server"""
         url = f"{self._base_url}{path}"
@@ -218,7 +215,7 @@ class ProxyRAGService(IRAGService):
         self,
         query: str,
         detected_libraries: Optional[List[str]] = None,
-        max_results: int = 5
+        max_results: int = 5,
     ) -> Optional[str]:
         """Get context via proxy"""
         logger.info(f"[Proxy] Get context for query: {query[:100]}...")

hdsp_agent_core/tests/conftest.py

@@ -5,7 +5,6 @@ Provides pytest fixtures for testing ServiceFactory and service implementations.
 """
 
 import os
-from typing import Generator
 from unittest.mock import AsyncMock, MagicMock, patch
 
 import pytest
@@ -15,6 +14,7 @@ import pytest
 def reset_factory():
     """Reset ServiceFactory singleton before each test"""
     from hdsp_agent_core.factory import ServiceFactory
+
     ServiceFactory.reset_instance()
     yield
     ServiceFactory.reset_instance()
@@ -76,6 +76,7 @@ def sample_plan_request():
     """Sample plan request for testing"""
     from hdsp_agent_core.models.agent import PlanRequest
     from hdsp_agent_core.models.common import NotebookContext
+
     return PlanRequest(
         request="Create a simple plot",
         notebookContext=NotebookContext(),
@@ -86,6 +87,7 @@ def sample_plan_request():
 def sample_chat_request():
     """Sample chat request for testing"""
     from hdsp_agent_core.models.chat import ChatRequest
+
     return ChatRequest(
         message="Hello, how can I analyze this data?",
         conversationId="test-conversation",
@@ -96,6 +98,7 @@ def sample_chat_request():
 def sample_search_request():
     """Sample search request for testing"""
     from hdsp_agent_core.models.rag import SearchRequest
+
     return SearchRequest(
         query="pandas dataframe operations",
         top_k=5,

hdsp_agent_core/tests/test_factory.py

@@ -137,7 +137,7 @@ class TestServiceFactoryInitialization:
         # Mock the RAG service initialization
         with patch(
             "hdsp_agent_core.services.rag_service.EmbeddedRAGService.initialize",
-            new_callable=AsyncMock
+            new_callable=AsyncMock,
         ):
             factory = ServiceFactory.get_instance()
             await factory.initialize()
@@ -241,7 +241,7 @@ class TestServiceFactoryModeSpecificServices:
         # Mock RAG initialization
         with patch(
             "hdsp_agent_core.services.rag_service.EmbeddedRAGService.initialize",
-            new_callable=AsyncMock
+            new_callable=AsyncMock,
         ):
             factory = ServiceFactory.get_instance()
             await factory.initialize()

hdsp_agent_core/tests/test_services.py

@@ -18,10 +18,8 @@ class TestProxyAgentService:
     def proxy_agent_service(self):
         """Create ProxyAgentService instance"""
         from hdsp_agent_core.services.agent_service import ProxyAgentService
-        return ProxyAgentService(
-            base_url="http://localhost:8000",
-            timeout=30.0
-        )
+
+        return ProxyAgentService(base_url="http://localhost:8000", timeout=30.0)
 
     def test_implements_interface(self, proxy_agent_service):
         """Test ProxyAgentService implements IAgentService"""
@@ -41,12 +39,8 @@ class TestProxyAgentService:
     ):
         """Test generate_plan makes HTTP POST request"""
         mock_response_data = {
-            "plan": {
-                "goal": "Create a simple plot",
-                "totalSteps": 1,
-                "steps": []
-            },
-            "reasoning": "Test reasoning"
+            "plan": {"goal": "Create a simple plot", "totalSteps": 1, "steps": []},
+            "reasoning": "Test reasoning",
         }
 
         with patch(
@@ -76,10 +70,8 @@ class TestProxyChatService:
     def proxy_chat_service(self):
         """Create ProxyChatService instance"""
         from hdsp_agent_core.services.chat_service import ProxyChatService
-        return ProxyChatService(
-            base_url="http://localhost:8000",
-            timeout=30.0
-        )
+
+        return ProxyChatService(base_url="http://localhost:8000", timeout=30.0)
 
     def test_implements_interface(self, proxy_chat_service):
         """Test ProxyChatService implements IChatService"""
@@ -119,10 +111,8 @@ class TestProxyRAGService:
     def proxy_rag_service(self):
         """Create ProxyRAGService instance"""
         from hdsp_agent_core.services.rag_service import ProxyRAGService
-        return ProxyRAGService(
-            base_url="http://localhost:8000",
-            timeout=30.0
-        )
+
+        return ProxyRAGService(base_url="http://localhost:8000", timeout=30.0)
 
     def test_implements_interface(self, proxy_rag_service):
         """Test ProxyRAGService implements IRAGService"""
@@ -189,6 +179,7 @@ class TestEmbeddedAgentService:
     def embedded_agent_service(self):
         """Create EmbeddedAgentService instance"""
         from hdsp_agent_core.services.agent_service import EmbeddedAgentService
+
         return EmbeddedAgentService()
 
     def test_implements_interface(self, embedded_agent_service):
@@ -202,7 +193,7 @@ class TestEmbeddedAgentService:
         """Test generate_plan calls LLM service"""
         # Mock the LLM service
         with patch.object(
-            embedded_agent_service, '_llm_service', create=True
+            embedded_agent_service, "_llm_service", create=True
         ) as mock_llm:
             mock_llm.generate_response = AsyncMock(
                 return_value='{"steps": [], "plan_id": "test"}'
@@ -219,6 +210,7 @@ class TestEmbeddedChatService:
     def embedded_chat_service(self):
         """Create EmbeddedChatService instance"""
         from hdsp_agent_core.services.chat_service import EmbeddedChatService
+
         return EmbeddedChatService()
 
     def test_implements_interface(self, embedded_chat_service):
@@ -233,6 +225,7 @@ class TestEmbeddedRAGService:
     def embedded_rag_service(self):
         """Create EmbeddedRAGService instance with mocked RAG manager"""
         from hdsp_agent_core.services.rag_service import EmbeddedRAGService
+
         service = EmbeddedRAGService()
         return service
 

{hdsp_jupyter_extension-2.0.10.data → hdsp_jupyter_extension-2.0.11.data}/data/share/jupyter/labextensions/hdsp-agent/build_log.json

@@ -722,7 +722,7 @@
             "@mui/material": {},
             "react-markdown": {},
             "hdsp-agent": {
-              "version": "2.0.10",
+              "version": "2.0.11",
               "singleton": true,
               "import": "/Users/a421721/Desktop/hdsp/hdsp_agent/extensions/jupyter/lib/index.js"
             }

{hdsp_jupyter_extension-2.0.10.data → hdsp_jupyter_extension-2.0.11.data}/data/share/jupyter/labextensions/hdsp-agent/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hdsp-agent",
-  "version": "2.0.10",
+  "version": "2.0.11",
   "description": "HDSP Agent JupyterLab Extension - Thin client for Agent Server",
   "keywords": [
     "jupyter",
@@ -127,7 +127,7 @@
       }
     },
     "_build": {
-      "load": "static/remoteEntry.4a252df3ade74efee8d6.js",
+      "load": "static/remoteEntry.9da31d1134a53b0c4af5.js",
       "extension": "./extension",
       "style": "./style"
     }

hdsp_jupyter_extension-2.0.10.data/data/share/jupyter/labextensions/hdsp-agent/static/lib_index_js.dc6434bee96ab03a0539.js → hdsp_jupyter_extension-2.0.11.data/data/share/jupyter/labextensions/hdsp-agent/static/lib_index_js.58c1e128ba0b76f41f04.js

@@ -3261,9 +3261,10 @@ SyntaxError: '(' was never closed
                     const interruptAction = msg.metadata?.interrupt?.action;
                     const isWriteFile = interruptAction === 'write_file_tool';
                     const isEditFile = interruptAction === 'edit_file_tool';
+                    const isMultiEditFile = interruptAction === 'multiedit_file_tool';
                     const writePath = (isWriteFile
                         && typeof msg.metadata?.interrupt?.args?.path === 'string') ? msg.metadata?.interrupt?.args?.path : '';
-                    const editPath = (isEditFile
+                    const editPath = ((isEditFile || isMultiEditFile)
                         && typeof msg.metadata?.interrupt?.args?.path === 'string') ? msg.metadata?.interrupt?.args?.path : '';
                     const autoApproved = msg.metadata?.interrupt?.autoApproved;
                     const headerRole = msg.role === 'user'
@@ -3283,10 +3284,26 @@ SyntaxError: '(' was never closed
                                 react__WEBPACK_IMPORTED_MODULE_0___default().createElement("div", { className: "jp-agent-interrupt-action-args" }, (() => {
                                     const command = msg.metadata?.interrupt?.args?.command;
                                     const code = msg.metadata?.interrupt?.args?.code || msg.metadata?.interrupt?.args?.content || '';
-                                    // Handle edit_file_tool with diff preview
+                                    // Handle edit_file_tool and multiedit_file_tool with diff preview
                                     let snippet;
                                     let language;
-                                    if (isEditFile) {
+                                    if (isMultiEditFile) {
+                                        // Handle multiedit_file_tool - show all edits
+                                        const edits = msg.metadata?.interrupt?.args?.edits || [];
+                                        const editsPreview = edits.slice(0, 5).map((edit, idx) => {
+                                            const oldStr = edit.old_string || '';
+                                            const newStr = edit.new_string || '';
+                                            const oldPreview = oldStr.length > 100 ? oldStr.slice(0, 100) + '...' : oldStr;
+                                            const newPreview = newStr.length > 100 ? newStr.slice(0, 100) + '...' : newStr;
+                                            return `# Edit ${idx + 1}${edit.replace_all ? ' (replace_all)' : ''}\n` +
+                                                oldPreview.split('\n').map((line) => `-${line}`).join('\n') + '\n' +
+                                                newPreview.split('\n').map((line) => `+${line}`).join('\n');
+                                        }).join('\n\n');
+                                        const moreEdits = edits.length > 5 ? `\n\n... and ${edits.length - 5} more edits` : '';
+                                        snippet = `--- ${editPath} (${edits.length} edits)\n+++ ${editPath} (after)\n\n${editsPreview}${moreEdits}`;
+                                        language = 'diff';
+                                    }
+                                    else if (isEditFile) {
                                         const oldStr = msg.metadata?.interrupt?.args?.old_string || '';
                                         const newStr = msg.metadata?.interrupt?.args?.new_string || '';
                                         const replaceAll = msg.metadata?.interrupt?.args?.replace_all;
@@ -3325,9 +3342,10 @@ SyntaxError: '(' was never closed
                                             const safePath = escapeHtml(writePath);
                                             html = html.replace(/<span class="code-block-language">[^<]*<\/span>/, `<span class="code-block-language jp-agent-interrupt-path">${safePath}</span>`);
                                         }
-                                        if (isEditFile && editPath) {
+                                        if ((isEditFile || isMultiEditFile) && editPath) {
                                             const safePath = escapeHtml(editPath);
-                                            html = html.replace(/<span class="code-block-language">[^<]*<\/span>/, `<span class="code-block-language jp-agent-interrupt-path">✏️ ${safePath}</span>`);
+                                            const editIcon = isMultiEditFile ? '📝' : '✏️';
+                                            html = html.replace(/<span class="code-block-language">[^<]*<\/span>/, `<span class="code-block-language jp-agent-interrupt-path">${editIcon} ${safePath}</span>`);
                                         }
                                         // actionHtml이 비어있지 않을 때만 추가
                                         return actionHtml ? html.replace('</div>', `${actionHtml}</div>`) : html;
@@ -9034,77 +9052,63 @@ __webpack_require__.r(__webpack_exports__);
  * - Key rotation on rate limit (429) is handled by frontend
  */
 const STORAGE_KEY = 'hdsp-agent-llm-config';
-const DEFAULT_LANGCHAIN_SYSTEM_PROMPT = `You are an expert Python data scientist and Jupyter notebook assistant.
-Your role is to help users with data analysis, visualization, and Python coding tasks in Jupyter notebooks. You can use only Korean
-
-# Core Behavior
-Be concise and direct. Answer in fewer than 4 lines unless the user asks for detail.
-After working on a file, just stop - don't explain what you did unless asked.
-Avoid unnecessary introductions or conclusions.
-
-## Task Management
-Use write_todos for complex multi-step tasks (3+ steps). Mark tasks in_progress before starting, completed immediately after finishing.
-For simple 1-2 step tasks, just do them directly without todos.
-When creating a todo list, ALWAYS include "작업 요약 및 다음단계 제시" as the LAST todo item.
-
-You MUST ALWAYS call a tool in every response. After any tool result, you MUST:
-1. Check your todo list - are there pending or in_progress items?
-2. If YES → call the next appropriate tool (jupyter_cell_tool, markdown_tool, etc.)
-3. When executing "작업 요약 및 다음단계 제시" (final todo):
-   - Output this JSON as text content: {"summary": "실행된 작업 요약", "next_items": [{"subject": "제목", "description": "설명"}]}
-   - AND call write_todos to mark all as 'completed' IN THE SAME RESPONSE
-   - Both content AND tool call must be in ONE response
-
-## 🔴 MANDATORY: Resource Check Before Data Hanlding
-**ALWAYS call check_resource_tool FIRST** when the task involves:
-- Loading files: .csv, .parquet, .json, .xlsx, .pickle, .h5, .feather
-- Handling datasets(dataframe) with pandas, polars, dask, or similar libraries
-- Training ML models on data files
-
-## Mandatory Workflow
-1. After EVERY tool result, immediately call the next tool
-2. Continue until ALL todos show status: "completed"
-3. When all todos are completed, the session ends automatically
-4. Only use jupyter_cell_tool for Python code or when the user explicitly asks to run in a notebook cell
-5. For plots and charts, use English text only.
-
-## ❌ FORBIDDEN (will break the workflow)
-- Producing an empty response (no tool call, no content)
-- Stopping after any tool without calling the next tool
-- Leaving todos in "in_progress" or "pending" state without continuing
-
-## 📂 File Search Best Practices
-**CRITICAL**: Use \`execute_command_tool\` with find/grep commands for file searching.
-
-**To find files by NAME** (e.g., find titanic.csv):
-- \`execute_command_tool(command="find . -iname '*titanic*.csv' 2>/dev/null")\`
-- \`execute_command_tool(command="find . -name '*.csv' 2>/dev/null")\` - find all CSV files
-
-**To search file CONTENTS** (e.g., find code containing "import pandas"):
-- \`execute_command_tool(command="grep -rn 'import pandas' --include='*.py' .")\`
-- \`execute_command_tool(command="grep -rn 'def train_model' --include='*.py' --include='*.ipynb' .")\`
-
-## 📖 File Reading Best Practices
-**CRITICAL**: When exploring codebases or reading files, use pagination to prevent context overflow.
-
-**Pattern for codebase exploration:**
-1. First scan: read_file_tool(path, limit=100) - See file structure and key sections
-2. Targeted read: read_file_tool(path, offset=100, limit=200) - Read specific sections if needed
-3. Full read: Only read without limit when necessary for immediate editing
-
-**When to paginate (use offset/limit):**
-- Reading any file >500 lines
-- Exploring unfamiliar codebases (always start with limit=100)
-- Reading multiple files in sequence
-- Any research or investigation task
-
-**When full read is OK:**
-- Small files (<500 lines)
-- Files you need to edit immediately after reading
-- After confirming file size with first scan
-
-## 🔧 Code Development
-For code generation/refactoring, use LSP tools (diagnostics_tool, references_tool) to check errors and find symbol usages. Use multiedit_file_tool for multiple changes in one file.
+const DEFAULT_LANGCHAIN_SYSTEM_PROMPT = `You are an expert Python data scientist and Jupyter notebook assistant. Respond in Korean only.
+
+# Core Rules
+1. Be concise (≤4 lines unless detail requested)
+2. ALWAYS call a tool in every response - never respond with text only
+3. ALWAYS include a brief Korean explanation before tool calls
+
+# Task Workflow
+
+## Simple Tasks (1-2 steps)
+Execute directly without todos.
+
+## Complex Tasks (3+ steps)
+1. Create todos with write_todos (all items in Korean)
+2. ALWAYS include "작업 요약 및 다음단계 제시" as the LAST item
+3. After each tool result: check todos → call next tool → repeat
+4. **Final todo ("작업 요약 및 다음단계 제시")**:
+   - FIRST: Output summary JSON in your content (REQUIRED!)
+   - THEN: Call write_todos to mark all as completed
+   - Both must be in the SAME response
+
+### Summary JSON Format (MUST output before marking complete)
+\`\`\`json
+{"summary": "실행된 작업 요약", "next_items": [{"subject": "제목", "description": "설명"}]}
+\`\`\`
+Suggest 3-5 next items. **You CANNOT mark "작업 요약" as completed without outputting this JSON first.**
+
+# Mandatory Checks
+
+## Resource Check (BEFORE data operations)
+Call \`check_resource_tool\` FIRST when:
+- Loading files (.csv, .parquet, .json, .xlsx, .pickle, .h5, .feather)
+- Using pandas/polars/dask for dataframes
+- Training ML models
+
+# Tool Usage
+
+## File Search (execute_command_tool)
+\`\`\`bash
+find . -iname '*filename*.csv' 2>/dev/null     # Find by name
+grep -rn 'pattern' --include='*.py' .           # Search contents
+\`\`\`
+
+## File Reading (read_file_tool)
+- Large files: \`read_file_tool(path, limit=100)\` first
+- Use \`offset\` for pagination
+- Small files (<500 lines): Read without limit
+
+## Code Output
+- For plots/charts: Use English labels only
+- Use LSP tools for error checking and symbol lookup
+- Use multiedit_file_tool for multiple changes
+
+# Forbidden
+- Empty responses (no tool call AND no content)
+- Tool calls without Korean explanation
+- Stopping with pending/in_progress todos
 `;
 // ═══════════════════════════════════════════════════════════════════════════
 // Key Rotation State (in-memory, not persisted)
@@ -16003,4 +16007,4 @@ __webpack_require__.r(__webpack_exports__);
 /***/ }
 
 }]);
-//# sourceMappingURL=lib_index_js.dc6434bee96ab03a0539.js.map
+//# sourceMappingURL=lib_index_js.58c1e128ba0b76f41f04.js.map