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

hdsp_agent_core/managers/config_manager.py

@@ -8,7 +8,7 @@ import json
 import os
 import tempfile
 from pathlib import Path
-from typing import Dict, Any
+from typing import Any, Dict
 
 
 class ConfigManager:
@@ -20,13 +20,13 @@ class ConfigManager:
     def __init__(self):
         try:
             # 1순위: 환경변수
-            jupyter_config_dir = os.environ.get('JUPYTER_CONFIG_DIR')
+            jupyter_config_dir = os.environ.get("JUPYTER_CONFIG_DIR")
 
             # 2순위: 홈 디렉토리 ~/.jupyter
             if not jupyter_config_dir:
                 # expanduser가 실패할 경우를 대비
                 try:
-                    jupyter_config_dir = os.path.expanduser('~/.jupyter')
+                    jupyter_config_dir = os.path.expanduser("~/.jupyter")
                 except Exception:
                     jupyter_config_dir = None
 
@@ -35,22 +35,22 @@ class ConfigManager:
                 config_path = Path(jupyter_config_dir)
                 try:
                     config_path.mkdir(parents=True, exist_ok=True)
-                    self._config_file = config_path / 'hdsp_agent_config.json'
+                    self._config_file = config_path / "hdsp_agent_config.json"
                 except Exception as e:
                     print(f"Warning: Cannot write to {jupyter_config_dir}: {e}")
                     self._config_file = None
 
             # 3순위 (비상용): 쓰기 실패했거나 경로가 없으면 /tmp 사용
             if not self._config_file:
-                tmp_dir = Path(tempfile.gettempdir()) / 'jupyter_hdsp_agent'
+                tmp_dir = Path(tempfile.gettempdir()) / "jupyter_hdsp_agent"
                 tmp_dir.mkdir(parents=True, exist_ok=True)
-                self._config_file = tmp_dir / 'hdsp_agent_config.json'
+                self._config_file = tmp_dir / "hdsp_agent_config.json"
                 print(f"Using temporary config path: {self._config_file}")
 
         except Exception as e:
             # 최악의 경우: 메모리에서만 동작하도록 가짜 경로 설정 (서버 다운 방지)
             print(f"Critical Error in ConfigManager init: {e}")
-            self._config_file = Path('/tmp/hdsp_agent_config_fallback.json')
+            self._config_file = Path("/tmp/hdsp_agent_config_fallback.json")
 
         self._config = self._load_config()
 
@@ -72,7 +72,7 @@ class ConfigManager:
             return self._default_config()
 
         try:
-            with open(self._config_file, 'r') as f:
+            with open(self._config_file, "r") as f:
                 return json.load(f)
         except (json.JSONDecodeError, IOError) as e:
             print(f"Error loading config: {e}")
@@ -81,20 +81,14 @@ class ConfigManager:
     def _default_config(self) -> Dict[str, Any]:
         """Get default configuration"""
         return {
-            'provider': 'gemini',
-            'gemini': {
-                'apiKey': '',
-                'model': 'gemini-2.5-pro'
+            "provider": "gemini",
+            "gemini": {"apiKey": "", "model": "gemini-2.5-pro"},
+            "vllm": {
+                "endpoint": "http://localhost:8000",
+                "apiKey": "",
+                "model": "meta-llama/Llama-2-7b-chat-hf",
             },
-            'vllm': {
-                'endpoint': 'http://localhost:8000',
-                'apiKey': '',
-                'model': 'meta-llama/Llama-2-7b-chat-hf'
-            },
-            'openai': {
-                'apiKey': '',
-                'model': 'gpt-4'
-            }
+            "openai": {"apiKey": "", "model": "gpt-4"},
         }
 
     def get_config(self) -> Dict[str, Any]:
@@ -113,7 +107,7 @@ class ConfigManager:
 
         # Write to file
         try:
-            with open(self._config_file, 'w') as f:
+            with open(self._config_file, "w") as f:
                 json.dump(self._config, f, indent=2)
         except IOError as e:
             raise RuntimeError(f"Failed to save config: {e}")

hdsp_agent_core/managers/session_manager.py

@@ -6,12 +6,12 @@ history to survive server restarts.
 """
 
 import json
+import logging
 import uuid
-from pathlib import Path
-from dataclasses import dataclass, asdict, field
-from typing import List, Dict, Optional
+from dataclasses import asdict, dataclass, field
 from datetime import datetime
-import logging
+from pathlib import Path
+from typing import Dict, List, Optional
 
 logger = logging.getLogger(__name__)
 
@@ -193,6 +193,7 @@ class SessionManager:
             # Lazy import to avoid circular dependency
             try:
                 from hdsp_agent_core.core.context_condenser import get_context_condenser
+
                 condenser = get_context_condenser()
                 msg_dicts = [{"role": m.role, "content": m.content} for m in messages]
                 compressed, _stats = condenser.condense(msg_dicts, target_tokens)

hdsp_agent_core/models/__init__.py

@@ -4,16 +4,6 @@ HDSP Agent Core - Pydantic Models
 Shared data models for the HDSP Agent system.
 """
 
-from .common import (
-    APIResponse,
-    ErrorInfo,
-    GeminiConfig,
-    LLMConfig,
-    NotebookContext,
-    OpenAIConfig,
-    ToolCall,
-    VLLMConfig,
-)
 from .agent import (
     DependencyInfo,
     ExecutionPlan,
@@ -22,14 +12,14 @@ from .agent import (
     PlanStep,
     RefineRequest,
     RefineResponse,
-    ReflectRequest,
-    ReflectResponse,
     ReflectionAdjustment,
     ReflectionAnalysis,
     ReflectionEvaluation,
     ReflectionImpact,
     ReflectionRecommendations,
     ReflectionResult,
+    ReflectRequest,
+    ReflectResponse,
     ReplanRequest,
     ReplanResponse,
     ReportExecutionRequest,
@@ -45,6 +35,16 @@ from .chat import (
     ChatResponse,
     StreamChunk,
 )
+from .common import (
+    APIResponse,
+    ErrorInfo,
+    GeminiConfig,
+    LLMConfig,
+    NotebookContext,
+    OpenAIConfig,
+    ToolCall,
+    VLLMConfig,
+)
 from .rag import (
     ChunkingConfig,
     EmbeddingConfig,

hdsp_agent_core/models/agent.py

@@ -155,9 +155,7 @@ class ReflectionImpact(BaseModel):
     affected_steps: List[int] = Field(
         default_factory=list, description="Step numbers affected"
     )
-    severity: str = Field(
-        description="Impact severity: none, minor, major, critical"
-    )
+    severity: str = Field(description="Impact severity: none, minor, major, critical")
     description: str = Field(description="Impact description")
 
 
@@ -169,13 +167,17 @@ class ReflectionAdjustment(BaseModel):
         description="Change type: modify_code, add_step, remove_step, change_approach"
     )
     description: str = Field(description="Adjustment description")
-    new_content: Optional[str] = Field(default=None, description="New content if applicable")
+    new_content: Optional[str] = Field(
+        default=None, description="New content if applicable"
+    )
 
 
 class ReflectionRecommendations(BaseModel):
     """Reflection recommendations"""
 
-    action: str = Field(description="Recommended action: continue, adjust, retry, replan")
+    action: str = Field(
+        description="Recommended action: continue, adjust, retry, replan"
+    )
     adjustments: List[ReflectionAdjustment] = Field(
         default_factory=list, description="Specific adjustments"
     )
@@ -187,7 +189,9 @@ class ReflectionResult(BaseModel):
 
     evaluation: ReflectionEvaluation = Field(description="Evaluation results")
     analysis: ReflectionAnalysis = Field(description="Analysis details")
-    impact_on_remaining: ReflectionImpact = Field(description="Impact on remaining steps")
+    impact_on_remaining: ReflectionImpact = Field(
+        description="Impact on remaining steps"
+    )
     recommendations: ReflectionRecommendations = Field(description="Recommendations")
 
 
@@ -199,7 +203,9 @@ class ReflectRequest(BaseModel):
     executedCode: str = Field(description="Code that was executed")
     executionStatus: str = Field(description="Execution status (success/error/warning)")
     executionOutput: str = Field(description="Output from execution")
-    errorMessage: Optional[str] = Field(default=None, description="Error message if any")
+    errorMessage: Optional[str] = Field(
+        default=None, description="Error message if any"
+    )
     expectedOutcome: Optional[str] = Field(
         default=None, description="Expected outcome description"
     )
@@ -297,7 +303,8 @@ class ValidateRequest(BaseModel):
 
     code: str = Field(description="Code to validate")
     notebookContext: Optional[NotebookContext] = Field(
-        default=None, description="Current notebook context (for variable/import tracking)"
+        default=None,
+        description="Current notebook context (for variable/import tracking)",
     )
 
 

hdsp_agent_core/models/common.py

@@ -60,8 +60,7 @@ class GeminiConfig(BaseModel):
 
     apiKey: str = Field(description="Primary Gemini API key")
     apiKeys: Optional[List[str]] = Field(
-        default=None,
-        description="Multiple API keys for rate limit rotation (max 10)"
+        default=None, description="Multiple API keys for rate limit rotation (max 10)"
     )
     model: str = Field(default="gemini-2.5-flash", description="Model name")
 

hdsp_agent_core/models/rag.py

@@ -10,7 +10,8 @@ Defines configuration schemas for the Local RAG system including:
 """
 
 import os
-from typing import Optional, List, Literal, Dict, Any
+from typing import Any, Dict, List, Literal, Optional
+
 from pydantic import BaseModel, Field
 
 
@@ -19,26 +20,22 @@ class QdrantConfig(BaseModel):
 
     mode: Literal["local", "server", "cloud"] = Field(
         default="local",
-        description="Deployment mode: local (file-based), server (Docker), cloud"
+        description="Deployment mode: local (file-based), server (Docker), cloud",
     )
     # Local mode settings
     local_path: Optional[str] = Field(
         default=None,
-        description="Path for local Qdrant storage. Defaults to ~/.hdsp_agent/qdrant"
+        description="Path for local Qdrant storage. Defaults to ~/.hdsp_agent/qdrant",
     )
     # Server mode settings
     url: Optional[str] = Field(
         default="http://localhost:6333",
-        description="Qdrant server URL for server/cloud mode"
-    )
-    api_key: Optional[str] = Field(
-        default=None,
-        description="API key for cloud mode"
+        description="Qdrant server URL for server/cloud mode",
     )
+    api_key: Optional[str] = Field(default=None, description="API key for cloud mode")
     # Collection settings
     collection_name: str = Field(
-        default="hdsp_knowledge",
-        description="Vector collection name"
+        default="hdsp_knowledge", description="Vector collection name"
     )
 
     def get_mode(self) -> str:
@@ -72,24 +69,20 @@ class EmbeddingConfig(BaseModel):
 
     model_name: str = Field(
         default="intfloat/multilingual-e5-small",
-        description="HuggingFace model name for embeddings"
+        description="HuggingFace model name for embeddings",
     )
     device: Literal["cpu", "cuda", "mps"] = Field(
-        default="cpu",
-        description="Device for embedding computation"
+        default="cpu", description="Device for embedding computation"
     )
     batch_size: int = Field(
-        default=32,
-        description="Batch size for embedding generation"
+        default=32, description="Batch size for embedding generation"
     )
     cache_folder: Optional[str] = Field(
-        default=None,
-        description="Model cache folder. Defaults to ~/.cache/huggingface"
+        default=None, description="Model cache folder. Defaults to ~/.cache/huggingface"
     )
     # Model-specific settings
     normalize_embeddings: bool = Field(
-        default=True,
-        description="Normalize embeddings for cosine similarity"
+        default=True, description="Normalize embeddings for cosine similarity"
     )
 
     def get_model_name(self) -> str:
@@ -104,59 +97,45 @@ class EmbeddingConfig(BaseModel):
 class ChunkingConfig(BaseModel):
     """Document chunking configuration"""
 
-    chunk_size: int = Field(
-        default=512,
-        description="Target chunk size in characters"
-    )
+    chunk_size: int = Field(default=512, description="Target chunk size in characters")
     chunk_overlap: int = Field(
-        default=50,
-        description="Overlap between chunks in characters"
+        default=50, description="Overlap between chunks in characters"
     )
     split_by_header: bool = Field(
-        default=True,
-        description="Split markdown by headers for semantic boundaries"
+        default=True, description="Split markdown by headers for semantic boundaries"
     )
     min_chunk_size: int = Field(
-        default=100,
-        description="Minimum chunk size to include"
+        default=100, description="Minimum chunk size to include"
     )
     max_chunk_size: int = Field(
-        default=2000,
-        description="Maximum chunk size (hard limit)"
+        default=2000, description="Maximum chunk size (hard limit)"
     )
 
 
 class WatchdogConfig(BaseModel):
     """File monitoring configuration"""
 
-    enabled: bool = Field(
-        default=True,
-        description="Enable file system monitoring"
-    )
+    enabled: bool = Field(default=True, description="Enable file system monitoring")
     debounce_seconds: float = Field(
-        default=2.0,
-        description="Debounce time for file change events"
+        default=2.0, description="Debounce time for file change events"
     )
     patterns: List[str] = Field(
         default=["*.md", "*.py", "*.txt", "*.json"],
-        description="File patterns to monitor"
+        description="File patterns to monitor",
     )
     ignore_patterns: List[str] = Field(
         default=[".*", "__pycache__", "*.pyc", "*.egg-info"],
-        description="Patterns to ignore"
+        description="Patterns to ignore",
     )
 
 
 class RAGConfig(BaseModel):
     """Main RAG system configuration"""
 
-    enabled: bool = Field(
-        default=True,
-        description="Enable RAG system"
-    )
+    enabled: bool = Field(default=True, description="Enable RAG system")
     knowledge_base_path: Optional[str] = Field(
         default=None,
-        description="Path to knowledge base. Defaults to built-in libraries/"
+        description="Path to knowledge base. Defaults to built-in libraries/",
     )
     qdrant: QdrantConfig = Field(default_factory=QdrantConfig)
     embedding: EmbeddingConfig = Field(default_factory=EmbeddingConfig)
@@ -164,17 +143,12 @@ class RAGConfig(BaseModel):
     watchdog: WatchdogConfig = Field(default_factory=WatchdogConfig)
 
     # Retrieval settings
-    top_k: int = Field(
-        default=5,
-        description="Number of chunks to retrieve"
-    )
+    top_k: int = Field(default=5, description="Number of chunks to retrieve")
     score_threshold: float = Field(
-        default=0.3,
-        description="Minimum similarity score threshold"
+        default=0.3, description="Minimum similarity score threshold"
     )
     max_context_tokens: int = Field(
-        default=1500,
-        description="Maximum tokens for RAG context injection"
+        default=1500, description="Maximum tokens for RAG context injection"
     )
 
     def is_enabled(self) -> bool:
@@ -203,35 +177,22 @@ class RAGConfig(BaseModel):
 class SearchRequest(BaseModel):
     """Request body for explicit RAG search"""
 
-    query: str = Field(
-        description="Search query"
-    )
-    top_k: Optional[int] = Field(
-        default=None,
-        description="Override default top_k"
-    )
+    query: str = Field(description="Search query")
+    top_k: Optional[int] = Field(default=None, description="Override default top_k")
     filters: Optional[Dict[str, Any]] = Field(
-        default=None,
-        description="Metadata filters (e.g., {'source_type': 'library'})"
+        default=None, description="Metadata filters (e.g., {'source_type': 'library'})"
     )
     include_score: bool = Field(
-        default=True,
-        description="Include relevance scores in results"
+        default=True, description="Include relevance scores in results"
     )
 
 
 class SearchResult(BaseModel):
     """Single search result"""
 
-    content: str = Field(
-        description="Chunk content"
-    )
-    score: float = Field(
-        description="Relevance score (0-1)"
-    )
-    metadata: Dict[str, Any] = Field(
-        description="Chunk metadata (source, type, etc.)"
-    )
+    content: str = Field(description="Chunk content")
+    score: float = Field(description="Relevance score (0-1)")
+    metadata: Dict[str, Any] = Field(description="Chunk metadata (source, type, etc.)")
 
 
 class SearchResponse(BaseModel):
@@ -240,35 +201,21 @@ class SearchResponse(BaseModel):
     results: List[SearchResult] = Field(
         description="Search results sorted by relevance"
     )
-    query: str = Field(
-        description="Original query"
-    )
-    total_results: int = Field(
-        description="Total matching chunks"
-    )
+    query: str = Field(description="Original query")
+    total_results: int = Field(description="Total matching chunks")
 
 
 class IndexStatusResponse(BaseModel):
     """Response body for index status"""
 
-    ready: bool = Field(
-        description="Whether RAG system is ready"
-    )
-    total_documents: int = Field(
-        default=0,
-        description="Total indexed documents"
-    )
-    total_chunks: int = Field(
-        default=0,
-        description="Total indexed chunks"
-    )
+    ready: bool = Field(description="Whether RAG system is ready")
+    total_documents: int = Field(default=0, description="Total indexed documents")
+    total_chunks: int = Field(default=0, description="Total indexed chunks")
     last_updated: Optional[str] = Field(
-        default=None,
-        description="Last index update timestamp"
+        default=None, description="Last index update timestamp"
     )
     knowledge_base_path: Optional[str] = Field(
-        default=None,
-        description="Current knowledge base path"
+        default=None, description="Current knowledge base path"
     )
 
 
@@ -276,30 +223,21 @@ class ReindexRequest(BaseModel):
     """Request body for manual reindexing"""
 
     force: bool = Field(
-        default=False,
-        description="Force full reindex even if files unchanged"
+        default=False, description="Force full reindex even if files unchanged"
     )
     path: Optional[str] = Field(
-        default=None,
-        description="Specific file or directory to reindex"
+        default=None, description="Specific file or directory to reindex"
     )
 
 
 class ReindexResponse(BaseModel):
     """Response body for reindex operation"""
 
-    success: bool = Field(
-        description="Whether reindex completed successfully"
-    )
-    indexed: int = Field(
-        description="Number of files indexed"
-    )
-    skipped: int = Field(
-        description="Number of files skipped (unchanged)"
-    )
+    success: bool = Field(description="Whether reindex completed successfully")
+    indexed: int = Field(description="Number of files indexed")
+    skipped: int = Field(description="Number of files skipped (unchanged)")
     errors: List[Dict[str, str]] = Field(
-        default=[],
-        description="List of indexing errors"
+        default=[], description="List of indexing errors"
     )
 
 
@@ -374,9 +312,8 @@ def get_default_rag_config() -> RAGConfig:
         enabled=True,
         qdrant=QdrantConfig(mode="local"),
         embedding=EmbeddingConfig(
-            model_name="intfloat/multilingual-e5-small",
-            device="cpu"
+            model_name="intfloat/multilingual-e5-small", device="cpu"
         ),
         chunking=ChunkingConfig(),
-        watchdog=WatchdogConfig()
+        watchdog=WatchdogConfig(),
     )

hdsp_agent_core/prompts/__init__.py

@@ -5,32 +5,32 @@ Prompt templates for Auto-Agent and Cell Actions.
 """
 
 from .auto_agent_prompts import (
-    PLAN_GENERATION_PROMPT,
+    ADAPTIVE_REPLAN_PROMPT,
     CODE_GENERATION_PROMPT,
+    ERROR_ANALYSIS_PROMPT,
     ERROR_REFINEMENT_PROMPT,
-    ADAPTIVE_REPLAN_PROMPT,
-    STRUCTURED_PLAN_PROMPT,
-    REFLECTION_PROMPT,
     FINAL_ANSWER_PROMPT,
-    ERROR_ANALYSIS_PROMPT,
     PIP_INDEX_OPTION,
+    PLAN_GENERATION_PROMPT,
+    REFLECTION_PROMPT,
+    STRUCTURED_PLAN_PROMPT,
+    format_error_analysis_prompt,
+    format_final_answer_prompt,
     format_plan_prompt,
     format_refine_prompt,
-    format_final_answer_prompt,
+    format_reflection_prompt,
     format_replan_prompt,
     format_structured_plan_prompt,
-    format_reflection_prompt,
-    format_error_analysis_prompt,
 )
 from .cell_action_prompts import (
-    EXPLAIN_CODE_PROMPT,
-    FIX_CODE_PROMPT,
     CUSTOM_REQUEST_PROMPT,
     DEFAULT_SYSTEM_PROMPT,
+    EXPLAIN_CODE_PROMPT,
+    FIX_CODE_PROMPT,
+    format_chat_prompt,
+    format_custom_prompt,
     format_explain_prompt,
     format_fix_prompt,
-    format_custom_prompt,
-    format_chat_prompt,
 )
 
 __all__ = [

hdsp_agent_core/prompts/cell_action_prompts.py

@@ -66,6 +66,7 @@ DEFAULT_SYSTEM_PROMPT = "당신은 도움이 되는 AI 어시스턴트입니다.
 # 유틸리티 함수
 # ═══════════════════════════════════════════════════════════════════════════
 
+
 def format_explain_prompt(cell_content: str) -> str:
     """코드 설명 프롬프트 포맷팅"""
     return EXPLAIN_CODE_PROMPT.format(cell_content=cell_content)
@@ -79,8 +80,7 @@ def format_fix_prompt(cell_content: str) -> str:
 def format_custom_prompt(custom_prompt: str, cell_content: str) -> str:
     """커스텀 요청 프롬프트 포맷팅"""
     return CUSTOM_REQUEST_PROMPT.format(
-        custom_prompt=custom_prompt,
-        cell_content=cell_content
+        custom_prompt=custom_prompt, cell_content=cell_content
     )
 
 
@@ -88,11 +88,13 @@ def format_chat_prompt(message: str, context: dict = None) -> str:
     """채팅 프롬프트 포맷팅"""
     prompt = message
 
-    if context and context.get('selectedCells'):
-        cells_text = '\n\n'.join([
-            f"셀 {i+1}:\n```python\n{cell}\n```"
-            for i, cell in enumerate(context['selectedCells'])
-        ])
+    if context and context.get("selectedCells"):
+        cells_text = "\n\n".join(
+            [
+                f"셀 {i + 1}:\n```python\n{cell}\n```"
+                for i, cell in enumerate(context["selectedCells"])
+            ]
+        )
         prompt = f"{message}\n\n노트북 컨텍스트:\n{cells_text}"
 
     return prompt