codegraph-engine 2.1.1__py3-none-any.whl

codegraph/llm/__init__.py

@@ -0,0 +1,3 @@
+from .base import LLMProvider, ProviderError
+
+__all__ = ["LLMProvider", "ProviderError"]

codegraph/llm/base.py

@@ -0,0 +1,15 @@
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+
+
+class ProviderError(RuntimeError):
+    """An external model provider could not complete a request."""
+
+
+class LLMProvider(ABC):
+    """Provider-neutral interface; CodeGraph works without a configured provider."""
+
+    @abstractmethod
+    async def complete(self, prompt: str, context: str) -> str:
+        raise NotImplementedError

codegraph/llm/context.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from codegraph.search import SearchResult
+
+
+def assemble_context(results: list[SearchResult], limit: int) -> str:
+    pieces: list[str] = []
+    used = 0
+    seen: set[tuple[str, int, int]] = set()
+    for result in results:
+        identity = (result.file, result.start_line, result.end_line)
+        if identity in seen:
+            continue
+        seen.add(identity)
+        piece = f"# {result.file}:{result.start_line}-{result.end_line}\n{result.snippet}\n"
+        if used + len(piece) > limit:
+            break
+        pieces.append(piece)
+        used += len(piece)
+    return "\n".join(pieces)

codegraph/mcp/__init__.py

@@ -0,0 +1,3 @@
+from .server import create_server
+
+__all__ = ["create_server"]