hdsp-jupyter-extension 2.0.27__py3-none-any.whl → 2.0.29__py3-none-any.whl

agent_server/config/__init__.py

@@ -0,0 +1,5 @@
+"""Agent Server Configuration Module"""
+
+from .server_config import ServerConfigManager, get_server_config_manager
+
+__all__ = ["ServerConfigManager", "get_server_config_manager"]

agent_server/config/server_config.py

@@ -0,0 +1,213 @@
+"""
+Server Configuration Manager - Agent Server specific settings
+
+This config manager handles all LLM and server settings.
+Stored in ~/.hdsp_agent/server_config.json
+
+This is separate from the client-side config which only stores agentServerUrl.
+"""
+
+import json
+import os
+import tempfile
+from pathlib import Path
+from typing import Any, Dict
+
+
+class ServerConfigManager:
+    """Manage Agent Server configuration persistence"""
+
+    _instance = None
+    _config_file = None
+
+    def __init__(self):
+        try:
+            # 1순위: 환경변수
+            config_dir = os.environ.get("HDSP_AGENT_CONFIG_DIR")
+
+            # 2순위: 홈 디렉토리 ~/.hdsp_agent
+            if not config_dir:
+                try:
+                    config_dir = os.path.expanduser("~/.hdsp_agent")
+                except Exception:
+                    config_dir = None
+
+            # 경로가 유효한지 체크하고, 없으면 생성 시도
+            if config_dir:
+                config_path = Path(config_dir)
+                try:
+                    config_path.mkdir(parents=True, exist_ok=True)
+                    self._config_file = config_path / "server_config.json"
+                except Exception as e:
+                    print(f"Warning: Cannot write to {config_dir}: {e}")
+                    self._config_file = None
+
+            # 3순위 (비상용): 쓰기 실패했거나 경로가 없으면 /tmp 사용
+            if not self._config_file:
+                tmp_dir = Path(tempfile.gettempdir()) / "hdsp_agent_server"
+                tmp_dir.mkdir(parents=True, exist_ok=True)
+                self._config_file = tmp_dir / "server_config.json"
+                print(f"Using temporary config path: {self._config_file}")
+
+        except Exception as e:
+            print(f"Critical Error in ServerConfigManager init: {e}")
+            self._config_file = Path("/tmp/hdsp_agent_server_config_fallback.json")
+
+        self._config = self._load_config()
+        print(f"ServerConfigManager initialized: {self._config_file}")
+
+    @classmethod
+    def get_instance(cls):
+        """Get singleton instance"""
+        if cls._instance is None:
+            cls._instance = ServerConfigManager()
+        return cls._instance
+
+    @classmethod
+    def reset_instance(cls):
+        """Reset singleton instance (for testing)"""
+        cls._instance = None
+
+    def _load_config(self) -> Dict[str, Any]:
+        """Load configuration from file"""
+        if not self._config_file.exists():
+            return self._default_config()
+
+        try:
+            with open(self._config_file, "r") as f:
+                loaded = json.load(f)
+                # Merge with defaults to ensure all keys exist
+                defaults = self._default_config()
+                defaults.update(loaded)
+                return defaults
+        except (json.JSONDecodeError, IOError) as e:
+            print(f"Error loading server config: {e}")
+            return self._default_config()
+
+    def _default_config(self) -> Dict[str, Any]:
+        """Get default server configuration"""
+        return {
+            # LLM Provider Settings
+            "provider": "gemini",
+            "gemini": {
+                "apiKey": "",
+                "apiKeys": [],
+                "model": "gemini-2.5-flash"
+            },
+            "openai": {
+                "apiKey": "",
+                "model": "gpt-4o"
+            },
+            "vllm": {
+                "endpoint": "https://openrouter.ai/api/v1",
+                "apiKey": "",
+                "model": "openai/gpt-4o",
+                "useResponsesApi": False
+            },
+            # Summarization LLM
+            "summarization": {
+                "enabled": False,
+                "provider": "gemini",
+                "model": None,
+                "endpoint": "",
+                "apiKey": ""
+            },
+            # Embedding
+            "embedding": {
+                "provider": "openai",
+                "model": "text-embedding-3-small",
+                "apiKey": "",
+                "endpoint": ""
+            },
+            # RAG
+            "rag": {
+                "qdrantUrl": "http://localhost:6333",
+                "collectionName": "hdsp_docs"
+            },
+            # Server Settings
+            "agentServerTimeout": 120.0,
+            "idleTimeout": 3600,  # 1 hour default
+            # Prompts (None = use defaults)
+            "prompts": {
+                "planner": None,
+                "python_developer": None,
+                "researcher": None,
+                "athena_query": None
+            },
+            # Feature Toggles
+            "useResponsesApi": False
+        }
+
+    def get_config(self) -> Dict[str, Any]:
+        """Get current configuration"""
+        self._config = self._load_config()
+        return self._config.copy()
+
+    def save_config(self, config: Dict[str, Any]):
+        """Save configuration to file"""
+        self._config.update(config)
+        self._config_file.parent.mkdir(parents=True, exist_ok=True)
+
+        try:
+            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 server config: {e}")
+
+    def get(self, key: str, default=None):
+        """Get specific config value"""
+        return self._config.get(key, default)
+
+    def set(self, key: str, value: Any):
+        """Set specific config value"""
+        self._config[key] = value
+        self.save_config(self._config)
+
+    def update_config(self, updates: Dict[str, Any]):
+        """Update configuration with partial updates (deep merge for nested dicts)"""
+        for key, value in updates.items():
+            if (
+                key in self._config
+                and isinstance(self._config[key], dict)
+                and isinstance(value, dict)
+            ):
+                self._config[key].update(value)
+            else:
+                self._config[key] = value
+        self.save_config(self._config)
+
+    def get_admin_config(self) -> Dict[str, Any]:
+        """Get all server configuration (for admin API)"""
+        config = self.get_config()
+        # Return everything except internal keys
+        return config.copy()
+
+    def update_admin_config(self, updates: Dict[str, Any]):
+        """Update server configuration"""
+        self.update_config(updates)
+
+    def get_user_config(self) -> Dict[str, Any]:
+        """Get user configuration.
+
+        NOTE: User preferences (workspaceRoot, autoApprove) are now stored
+        in browser localStorage. This method returns defaults for backward
+        compatibility with the API.
+        """
+        return {
+            "workspaceRoot": "",
+            "temperature": 0.7,
+            "autoApprove": False
+        }
+
+    def update_user_config(self, updates: Dict[str, Any]):
+        """Update user configuration.
+
+        NOTE: User preferences are stored in browser localStorage.
+        This is a no-op for backward compatibility.
+        """
+        pass  # No-op: user config is in localStorage
+
+
+def get_server_config_manager() -> ServerConfigManager:
+    """Get the singleton ServerConfigManager instance"""
+    return ServerConfigManager.get_instance()

agent_server/context_providers/__init__.py

@@ -2,11 +2,11 @@
 Context Providers Module
 
 Provides context injection for LLM prompts (e.g., @file, @notebook, @selection).
-Also provides action commands (e.g., @reset).
+Also provides action commands (e.g., /reset).
 Inspired by jupyter-ai's context provider architecture.
 """
 
-from .actions import ResetContextProvider
+from .actions import CompactContextProvider, HelpContextProvider, ResetContextProvider
 from .base import BaseContextProvider, ContextCommand, find_commands
 from .file import FileContextProvider
 from .processor import ContextProcessor, process_context_commands
@@ -16,6 +16,8 @@ __all__ = [
     "BaseContextProvider",
     "FileContextProvider",
     "ResetContextProvider",
+    "CompactContextProvider",
+    "HelpContextProvider",
     "ContextProcessor",
     "find_commands",
     "process_context_commands",

agent_server/context_providers/actions.py

@@ -1,7 +1,7 @@
 """
 Action Context Providers
 
-These providers handle action commands like @reset that perform actions
+These providers handle action commands like /reset that perform actions
 rather than providing context to the LLM.
 """
 
@@ -13,11 +13,11 @@ class ResetContextProvider(BaseContextProvider):
     Action provider for resetting the session.
 
     Usage:
-        @reset - Clear conversation history and recreate agent
+        /reset - Clear conversation history and recreate agent
     """
 
-    id = "@reset"
-    description = "Clear session and reset agent"
+    id = "/reset"
+    description = "세션 초기화 및 에이전트 리셋"
     requires_arg = False
     only_start = False
 
@@ -25,15 +25,15 @@ class ResetContextProvider(BaseContextProvider):
         super().__init__(base_dir)
 
     def get_arg_options(self, arg_prefix: str) -> list[ListOptionsEntry]:
-        """No arguments for @reset."""
+        """No arguments for /reset."""
         return []
 
     def make_context(self, command: ContextCommand) -> str:
-        """@reset doesn't provide context - it's an action command."""
+        """/reset doesn't provide context - it's an action command."""
         return ""
 
     def replace_command(self, command: ContextCommand) -> str:
-        """Replace @reset with empty string (it's an action, not context)."""
+        """Replace /reset with empty string (it's an action, not context)."""
         return ""
 
     def is_action_command(self) -> bool:
@@ -43,3 +43,69 @@ class ResetContextProvider(BaseContextProvider):
 
 # Add more action commands here as needed
 # Example: @clear, @help, @history, etc.
+
+
+class CompactContextProvider(BaseContextProvider):
+    """
+    Action provider for compacting conversation history.
+
+    Usage:
+        /compact - Summarize older messages, keep recent 3 intact
+    """
+
+    id = "/compact"
+    description = "대화 기록 압축 및 요약"
+    requires_arg = False
+    only_start = False
+
+    def __init__(self, base_dir: str = "."):
+        super().__init__(base_dir)
+
+    def get_arg_options(self, arg_prefix: str) -> list[ListOptionsEntry]:
+        """No arguments for /compact."""
+        return []
+
+    def make_context(self, command: ContextCommand) -> str:
+        """/compact doesn't provide context - it's an action command."""
+        return ""
+
+    def replace_command(self, command: ContextCommand) -> str:
+        """Replace /compact with empty string (it's an action, not context)."""
+        return ""
+
+    def is_action_command(self) -> bool:
+        """Indicate this is an action command."""
+        return True
+
+
+class HelpContextProvider(BaseContextProvider):
+    """
+    Action provider for displaying help information.
+
+    Usage:
+        /help - Show available commands and features guide
+    """
+
+    id = "/help"
+    description = "도움말 및 사용 가이드"
+    requires_arg = False
+    only_start = False
+
+    def __init__(self, base_dir: str = "."):
+        super().__init__(base_dir)
+
+    def get_arg_options(self, arg_prefix: str) -> list[ListOptionsEntry]:
+        """No arguments for /help."""
+        return []
+
+    def make_context(self, command: ContextCommand) -> str:
+        """/help doesn't provide context - it's an action command."""
+        return ""
+
+    def replace_command(self, command: ContextCommand) -> str:
+        """Replace /help with empty string (it's an action, not context)."""
+        return ""
+
+    def is_action_command(self) -> bool:
+        """Indicate this is an action command."""
+        return True

agent_server/context_providers/file.py

@@ -73,7 +73,7 @@ class FileContextProvider(BaseContextProvider):
     """
 
     id = "@file"
-    description = "Include file contents in the prompt"
+    description = "파일 내용을 프롬프트에 포함"
     requires_arg = True
     only_start = False
 
@@ -152,7 +152,7 @@ class FileContextProvider(BaseContextProvider):
                 ListOptionsEntry(
                     id=self.command_id,
                     label=f"{self.command_id}:{label_path}",
-                    description="Directory"
+                    description="디렉토리"
                     if is_dir
                     else self._get_file_description(path),
                     only_start=self.only_start,
@@ -168,26 +168,26 @@ class FileContextProvider(BaseContextProvider):
         basename = os.path.basename(filepath)
 
         descriptions = {
-            ".py": "Python file",
-            ".ipynb": "Jupyter notebook",
-            ".md": "Markdown file",
-            ".txt": "Text file",
-            ".json": "JSON file",
-            ".yaml": "YAML file",
-            ".yml": "YAML file",
-            ".toml": "TOML file",
-            ".js": "JavaScript file",
-            ".ts": "TypeScript file",
-            ".tsx": "TypeScript React file",
-            ".jsx": "JavaScript React file",
-            ".html": "HTML file",
-            ".css": "CSS file",
-            ".sql": "SQL file",
-            ".sh": "Shell script",
-            ".csv": "CSV file",
-            ".r": "R file",
-            ".R": "R file",
-            ".jl": "Julia file",
+            ".py": "Python 파일",
+            ".ipynb": "Jupyter 노트북",
+            ".md": "Markdown 파일",
+            ".txt": "텍스트 파일",
+            ".json": "JSON 파일",
+            ".yaml": "YAML 파일",
+            ".yml": "YAML 파일",
+            ".toml": "TOML 파일",
+            ".js": "JavaScript 파일",
+            ".ts": "TypeScript 파일",
+            ".tsx": "TypeScript React 파일",
+            ".jsx": "JavaScript React 파일",
+            ".html": "HTML 파일",
+            ".css": "CSS 파일",
+            ".sql": "SQL 파일",
+            ".sh": "셸 스크립트",
+            ".csv": "CSV 파일",
+            ".r": "R 파일",
+            ".R": "R 파일",
+            ".jl": "Julia 파일",
         }
 
         # Check special filenames
@@ -196,7 +196,7 @@ class FileContextProvider(BaseContextProvider):
         if basename == "Makefile":
             return "Makefile"
 
-        return descriptions.get(ext, "File")
+        return descriptions.get(ext, "파일")
 
     def make_context(self, command: ContextCommand) -> str:
         """

agent_server/core/__init__.py

@@ -2,7 +2,7 @@
 Core services for HDSP Agent Server
 """
 
-from hdsp_agent_core.managers.config_manager import ConfigManager
+from agent_server.config import ServerConfigManager
 from hdsp_agent_core.managers.session_manager import (
     ChatMessage,
     Session,
@@ -47,7 +47,7 @@ from .state_verifier import (
 from .summary_generator import SummaryGenerator, TaskType, get_summary_generator
 
 __all__ = [
-    "ConfigManager",
+    "ServerConfigManager",
     "LLMClient",
     "LLMService",
     "PromptBuilder",

agent_server/core/llm_service.py

@@ -28,11 +28,10 @@ class LLMService:
             return self._key_manager
         if self.provider == "gemini":
             try:
-                from hdsp_agent_core.managers.config_manager import ConfigManager
-
+                from agent_server.config import ServerConfigManager
                 from agent_server.core.api_key_manager import get_key_manager
 
-                return get_key_manager(ConfigManager.get_instance())
+                return get_key_manager(ServerConfigManager.get_instance())
             except ImportError:
                 # Fallback for standalone usage
                 return None

agent_server/langchain/__init__.py

@@ -12,7 +12,7 @@ Architecture:
 - state.py: Agent state management
 """
 
-from agent_server.langchain.agent import create_simple_chat_agent, create_agent_system
+from agent_server.langchain.agent import create_agent_system
 from agent_server.langchain.state import AgentState
 
-__all__ = ["create_simple_chat_agent", "create_agent_system", "AgentState"]
+__all__ = ["create_agent_system", "AgentState"]