codebuff-agent 2.1.74__py3-none-any.whl

codebuff_core/directory_policy.py

@@ -0,0 +1,32 @@
+"""Directory exclusions shared by workspace consumers.
+
+Keep purpose-specific generated-output exclusions explicit: AST browsing may
+inspect build output, while autocomplete should not suggest bundled JS files.
+"""
+
+COMMON_EXCLUDED_DIRS = frozenset(
+    {
+        "__pycache__",
+        ".git",
+        "node_modules",
+        ".venv",
+        "venv",
+        ".tox",
+        ".mypy_cache",
+        ".codebuff-home",  # 워크트리 서버의 Qdrant·세션 (scripts/jl_env.sh, 이슈 #116)
+    }
+)
+GENERATED_EXCLUDED_DIRS = COMMON_EXCLUDED_DIRS | {
+    ".ipynb_checkpoints",
+    ".pytest_cache",
+    ".ruff_cache",
+    "dist",
+    "build",
+    "target",  # Maven·Gradle(Kotlin)·Rust 빌드 산출물 (이슈 #100)
+}
+SEARCH_EXCLUDED_DIRS = GENERATED_EXCLUDED_DIRS | {".env", ".eggs"}
+FILE_INDEX_EXCLUDED_DIRS = GENERATED_EXCLUDED_DIRS | {".eggs", "labextension", "lib"}
+AST_EXCLUDED_DIRS = COMMON_EXCLUDED_DIRS
+
+# Relative to the requested search root, so an explicit worktree scope still works.
+EXCLUDED_RELATIVE_DIRS = frozenset({(".claude", "worktrees")})

codebuff_core/factory.py

@@ -0,0 +1,155 @@
+"""
+Codebuff Core - 서비스 팩토리
+
+Embedded 서비스 구현을 생성하는 팩토리.
+"""
+
+import logging
+from typing import Optional
+
+from codebuff_core.interfaces import IAgentService, IChatService, IRAGService
+
+logger = logging.getLogger(__name__)
+
+
+class ServiceFactory:
+    """
+    서비스 인스턴스를 생성하는 팩토리.
+
+    싱글톤 패턴으로 애플리케이션 전체에서 일관된 서비스 인스턴스를 보장함.
+
+    사용법:
+        factory = ServiceFactory.get_instance()
+        await factory.initialize()
+
+        agent_service = factory.get_agent_service()
+        chat_service = factory.get_chat_service()
+        rag_service = factory.get_rag_service()
+    """
+
+    _instance: Optional["ServiceFactory"] = None
+
+    def __init__(self):
+        """팩토리 초기화 (get_instance() 사용 권장)"""
+        self._initialized = False
+
+        # 서비스 인스턴스 (지연 로딩)
+        self._agent_service: Optional[IAgentService] = None
+        self._chat_service: Optional[IChatService] = None
+        self._rag_service: Optional[IRAGService] = None
+
+        logger.info("ServiceFactory created")
+
+    @classmethod
+    def get_instance(cls) -> "ServiceFactory":
+        """싱글톤 팩토리 인스턴스 반환"""
+        if cls._instance is None:
+            cls._instance = ServiceFactory()
+        return cls._instance
+
+    @classmethod
+    def reset_instance(cls) -> None:
+        """싱글톤 인스턴스 초기화 (테스트용)"""
+        cls._instance = None
+
+    @property
+    def is_initialized(self) -> bool:
+        """팩토리 초기화 여부 확인"""
+        return self._initialized
+
+    async def initialize(self) -> None:
+        """
+        모든 서비스 초기화.
+
+        임베디드 서비스: RAG 매니저 초기화, 지식 베이스 로드
+        """
+        if self._initialized:
+            logger.debug("ServiceFactory already initialized")
+            return
+
+        logger.info("Initializing ServiceFactory")
+
+        from codebuff_core.services.agent_service import EmbeddedAgentService
+        from codebuff_core.services.chat_service import EmbeddedChatService
+        from codebuff_core.services.rag_service import EmbeddedRAGService
+
+        # 서비스 인스턴스 생성
+        self._agent_service = EmbeddedAgentService()
+        self._chat_service = EmbeddedChatService()
+        self._rag_service = EmbeddedRAGService()
+
+        # RAG 서비스 초기화 (비동기 초기화 필요할 수 있음)
+        await self._rag_service.initialize()
+
+        self._initialized = True
+        logger.info("ServiceFactory initialization complete")
+
+    async def shutdown(self) -> None:
+        """모든 서비스 종료 및 리소스 정리"""
+        logger.info("Shutting down ServiceFactory")
+
+        # RAG 서비스 정리
+        if self._rag_service and hasattr(self._rag_service, "shutdown"):
+            await self._rag_service.shutdown()
+
+        # 서비스 인스턴스 초기화
+        self._agent_service = None
+        self._chat_service = None
+        self._rag_service = None
+        self._initialized = False
+
+        logger.info("ServiceFactory shutdown complete")
+
+    def get_agent_service(self) -> IAgentService:
+        """
+        Agent 서비스 인스턴스 반환.
+
+        Returns:
+            IAgentService 구현체
+
+        Raises:
+            RuntimeError: 팩토리가 초기화되지 않은 경우
+        """
+        if not self._initialized:
+            raise RuntimeError(
+                "ServiceFactory not initialized. Call await factory.initialize() first."
+            )
+        return self._agent_service
+
+    def get_chat_service(self) -> IChatService:
+        """
+        Chat 서비스 인스턴스 반환.
+
+        Returns:
+            IChatService 구현체
+
+        Raises:
+            RuntimeError: 팩토리가 초기화되지 않은 경우
+        """
+        if not self._initialized:
+            raise RuntimeError(
+                "ServiceFactory not initialized. Call await factory.initialize() first."
+            )
+        return self._chat_service
+
+    def get_rag_service(self) -> IRAGService:
+        """
+        RAG 서비스 인스턴스 반환.
+
+        Returns:
+            IRAGService 구현체
+
+        Raises:
+            RuntimeError: 팩토리가 초기화되지 않은 경우
+        """
+        if not self._initialized:
+            raise RuntimeError(
+                "ServiceFactory not initialized. Call await factory.initialize() first."
+            )
+        return self._rag_service
+
+
+# 싱글톤 팩토리 편의 함수
+def get_service_factory() -> ServiceFactory:
+    """싱글톤 ServiceFactory 인스턴스 반환"""
+    return ServiceFactory.get_instance()

codebuff_core/interfaces.py

@@ -0,0 +1,197 @@
+"""
+Codebuff Core - 서비스 인터페이스
+
+에이전트 서비스의 계약을 정의하는 추상 인터페이스.
+"""
+
+from abc import ABC, abstractmethod
+from typing import Any, AsyncGenerator, Dict, List, Optional
+
+from codebuff_core.models.agent import (
+    PlanRequest,
+    PlanResponse,
+    RefineRequest,
+    RefineResponse,
+    ReplanRequest,
+    ReplanResponse,
+)
+from codebuff_core.models.chat import ChatRequest, ChatResponse
+from codebuff_core.models.rag import SearchRequest, SearchResponse
+
+
+class IAgentService(ABC):
+    """
+    에이전트 서비스의 추상 인터페이스.
+
+    계획 생성, 코드 개선, 재계획 작업을 처리합니다.
+    EmbeddedAgentService로 구현됩니다.
+    """
+
+    @abstractmethod
+    async def generate_plan(self, request: PlanRequest) -> PlanResponse:
+        """
+        자연어 요청으로부터 실행 계획을 생성합니다.
+
+        Args:
+            request: 사용자 요청과 노트북 컨텍스트가 포함된 계획 생성 요청
+
+        Returns:
+            구조화된 실행 계획이 담긴 PlanResponse
+        """
+        ...
+
+    @abstractmethod
+    async def refine_code(self, request: RefineRequest) -> RefineResponse:
+        """
+        실행 오류 발생 후 코드를 개선합니다.
+
+        Args:
+            request: 실패한 단계와 오류 정보가 포함된 개선 요청
+
+        Returns:
+            수정된 도구 호출이 담긴 RefineResponse
+        """
+        ...
+
+    @abstractmethod
+    async def replan(self, request: ReplanRequest) -> ReplanResponse:
+        """
+        실패한 단계를 어떻게 처리할지 결정합니다.
+
+        오류 분류를 사용하여 복구 전략을 결정합니다.
+
+        Args:
+            request: 오류 세부사항과 실행 컨텍스트가 포함된 재계획 요청
+
+        Returns:
+            결정과 변경사항이 담긴 ReplanResponse
+        """
+        ...
+
+    @abstractmethod
+    async def validate_code(
+        self, code: str, notebook_context: Optional[Dict] = None
+    ) -> Dict[str, Any]:
+        """
+        실행 전 코드를 검증합니다.
+
+        Args:
+            code: 검증할 Python 코드
+            notebook_context: 검증을 위한 선택적 노트북 컨텍스트
+
+        Returns:
+            이슈와 의존성이 포함된 검증 결과
+        """
+        ...
+
+
+class IChatService(ABC):
+    """
+    채팅 서비스의 추상 인터페이스.
+
+    LLM과의 대화형 상호작용을 처리합니다.
+    EmbeddedChatService로 구현됩니다.
+    """
+
+    @abstractmethod
+    async def send_message(self, request: ChatRequest) -> ChatResponse:
+        """
+        채팅 메시지를 보내고 응답을 받습니다.
+
+        Args:
+            request: 메시지와 선택적 대화 ID가 포함된 채팅 요청
+
+        Returns:
+            LLM 응답과 대화 ID가 담긴 ChatResponse
+        """
+        ...
+
+    @abstractmethod
+    async def send_message_stream(
+        self, request: ChatRequest
+    ) -> AsyncGenerator[Dict[str, Any], None]:
+        """
+        채팅 메시지를 보내고 스트리밍 응답을 받습니다.
+
+        Args:
+            request: 메시지와 선택적 대화 ID가 포함된 채팅 요청
+
+        Yields:
+            부분 콘텐츠가 포함된 스트림 청크
+        """
+        ...
+
+
+class IRAGService(ABC):
+    """
+    RAG 서비스의 추상 인터페이스.
+
+    검색 증강 생성 작업을 처리합니다.
+    EmbeddedRAGService로 구현됩니다.
+    """
+
+    @abstractmethod
+    async def search(self, request: SearchRequest) -> SearchResponse:
+        """
+        지식 베이스를 검색합니다.
+
+        Args:
+            request: 쿼리와 선택적 필터가 포함된 검색 요청
+
+        Returns:
+            일치하는 문서와 점수가 담긴 SearchResponse
+        """
+        ...
+
+    @abstractmethod
+    async def get_context_for_query(
+        self,
+        query: str,
+        detected_libraries: Optional[List[str]] = None,
+        max_results: int = 5,
+    ) -> Optional[str]:
+        """
+        쿼리에 대한 포맷된 컨텍스트를 가져옵니다 (프롬프트 주입용).
+
+        Args:
+            query: 사용자 쿼리
+            detected_libraries: 쿼리에서 감지된 라이브러리들
+            max_results: 포함할 최대 결과 수
+
+        Returns:
+            포맷된 컨텍스트 문자열 또는 사용 불가시 None
+        """
+        ...
+
+    @abstractmethod
+    def is_ready(self) -> bool:
+        """
+        RAG 서비스가 준비되었는지 확인합니다.
+
+        Returns:
+            서비스가 초기화되어 준비되면 True
+        """
+        ...
+
+    @abstractmethod
+    async def get_index_status(self) -> Dict[str, Any]:
+        """
+        현재 인덱스 상태를 가져옵니다.
+
+        Returns:
+            문서 수, 컬렉션 등이 포함된 인덱스 상태
+        """
+        ...
+
+    @abstractmethod
+    async def trigger_reindex(self, force: bool = False) -> Dict[str, Any]:
+        """
+        재인덱싱 작업을 트리거합니다.
+
+        Args:
+            force: True이면 필요하지 않아도 전체 재인덱싱 강제 실행
+
+        Returns:
+            재인덱싱 작업 상태
+        """
+        ...

codebuff_core/knowledge/__init__.py

@@ -0,0 +1,31 @@
+"""
+Codebuff Core - Knowledge Base
+
+Deterministic library detection and API guide management.
+"""
+
+from .chunking import (
+    DocumentChunker,
+    chunk_file,
+)
+from .loader import (
+    LIBRARY_DESCRIPTIONS,
+    KnowledgeBase,
+    KnowledgeLoader,
+    LibraryDetector,
+    get_knowledge_base,
+    get_knowledge_loader,
+    get_library_detector,
+)
+
+__all__ = [
+    "KnowledgeBase",
+    "KnowledgeLoader",
+    "LibraryDetector",
+    "get_knowledge_base",
+    "get_knowledge_loader",
+    "get_library_detector",
+    "LIBRARY_DESCRIPTIONS",
+    "DocumentChunker",
+    "chunk_file",
+]