hdsp-jupyter-extension 2.0.8__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.8.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.8",
+              "version": "2.0.11",
               "singleton": true,
               "import": "/Users/a421721/Desktop/hdsp/hdsp_agent/extensions/jupyter/lib/index.js"
             }

{hdsp_jupyter_extension-2.0.8.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.8",
+  "version": "2.0.11",
   "description": "HDSP Agent JupyterLab Extension - Thin client for Agent Server",
   "keywords": [
     "jupyter",
@@ -127,7 +127,7 @@
       }
     },
     "_build": {
-      "load": "static/remoteEntry.020cdb0b864cfaa4e41e.js",
+      "load": "static/remoteEntry.9da31d1134a53b0c4af5.js",
       "extension": "./extension",
       "style": "./style"
     }