awgraph 1.0.0__py3-none-any.whl

awgraph/__init__.py

@@ -0,0 +1,70 @@
+"""awgraph — Semantic Python Code Graph.
+
+A lightweight, portable Python AST indexer with call graphs, embeddings support,
+and hybrid keyword+semantic search. Originally developed as part of AitherOS's
+CodeGraph faculty, now available as a public package.
+
+## What it does
+
+- Real AST parsing (not regex)
+- Extracts functions, classes, and methods into chunks
+- Builds call graphs (what calls what, what's called by what)
+- Hybrid search: keyword (FTS5) + semantic (embeddings)
+- SQLite storage with WAL mode for concurrent access
+- Optional NumPy for fast vector operations
+
+## What it doesn't do
+
+- PDFs, web pages, or "universal" anything — just Python code
+- Embeddings by default — bring your own or use plugin hooks
+- Network access by default — embeddings API must be provided externally
+- Persistence across instances — in-memory by design; SQLite for durability
+
+## Usage
+
+```python
+from awgraph import CodeGraph
+
+# Create an indexer
+graph = CodeGraph()
+
+# Index a directory
+await graph.index_codebase("/path/to/code")
+
+# Query
+chunks = await graph.query("rate limiter", limit=10)
+
+for chunk in chunks:
+    print(f"{chunk.name}: {chunk.calls}")
+```
+
+## Architecture
+
+- **graph.py**: CodeGraph engine, AST parsing, call graphs, chunking
+- **store.py**: SQLite+FTS5 backend for storage and search
+- **registry.py**: Multi-root manager for indexing external repos
+- **base.py**: Minimal faculty graph base class
+- **logging.py**: Standard Python logging shim
+- **degradation.py**: Optional dependency tracking
+
+Plugin support via awgraph.plugins (not yet public).
+"""
+
+from __future__ import annotations
+
+from awgraph.graph import CodeGraph, CodeChunk
+from awgraph.registry import CodeGraphRegistry, get_codegraph_registry
+from awgraph.store import CodeGraphStore
+from awgraph.base import BaseFacultyGraph, GraphSyncConfig
+
+__version__ = "1.0.0"
+
+__all__ = [
+    "CodeGraph",
+    "CodeChunk",
+    "CodeGraphRegistry",
+    "CodeGraphStore",
+    "BaseFacultyGraph",
+    "GraphSyncConfig",
+    "get_codegraph_registry",
+]

awgraph/base.py

@@ -0,0 +1,139 @@
+"""
+Minimal base class for awgraph.
+
+Provides the core interface a faculty graph implements, without the AitherOS
+integrations (sync, provenance, integrity). This is what the public package
+exposes.
+"""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import Any, Dict, List
+
+
+@dataclass
+class GraphSyncConfig:
+    """Stub for compatibility; sync is not available in the public package."""
+    enabled: bool = False
+    domain: str = ""
+    batch_size: int = 20
+    flush_interval: float = 5.0
+    source_graph: str = ""
+    provenance: bool = False
+    provenance_node_type: str = "claim"
+
+
+class BaseFacultyGraph:
+    """
+    Minimal base class for a faculty graph.
+
+    In AitherOS, this also provides sync to AitherKnowledgeGraph, provenance
+    emission, and integrity tracking. The public package drops those features,
+    keeping only the core interface: query(), stats(), and graph_query().
+    """
+
+    _sync_config: GraphSyncConfig = GraphSyncConfig()
+    _scope_level: str = "platform"
+
+    def __init__(self):
+        self._sync_config = GraphSyncConfig()
+
+    def _queue_sync(self, node_data: Dict[str, Any], tenant_id: str = "platform") -> None:
+        """No-op: sync to AitherKnowledgeGraph is not available in public package."""
+        pass
+
+    def _queue_deletion(self, node_id: str, tenant_id: str = "platform") -> None:
+        """No-op: sync is not available in public package."""
+        pass
+
+    # SYNC on purpose: every call site is `self._flush_to_bus()` with no await.
+    # Declaring it async made each call return a coroutine nobody awaits, which
+    # Python reports only as a RuntimeWarning — so the no-op would have looked
+    # like it ran while doing nothing, in a package whose whole promise is that
+    # the stub is inert rather than silently broken.
+    def _flush_to_bus(self, *args: Any, **kwargs: Any) -> None:
+        """No-op: there is no event bus in the standalone package.
+
+        This one is not decoration. Indexing calls it, so omitting it made the
+        package import cleanly and then die with AttributeError on the FIRST
+        index_codebase() call — the shape where a vendored package looks fine to
+        every import check and is broken for the person who pip-installed it.
+        An import test cannot catch this; only actually indexing something can.
+
+        Subclasses in a host application may override it to re-attach their own
+        event plumbing; nothing here needs to know that they exist.
+        """
+        return None
+
+    def _emit_provenance(
+        self,
+        node_id: str,
+        name: str,
+        properties: Dict[str, Any],
+        tenant_id: str = "platform",
+    ) -> None:
+        """No-op: provenance emission is not available in public package."""
+        pass
+
+    def graph_stats(self) -> Dict[str, Any]:
+        """Inventory: how many nodes does this graph hold?
+
+        Subclasses may override. Default derives counts from public attributes.
+        """
+        entities: Dict[str, int] = {}
+        for name, val in vars(self).items():
+            if name.startswith("_") or name.startswith("by_"):
+                continue
+            if not isinstance(val, (dict, list, set, tuple)):
+                continue
+            entities[name] = len(val)
+        return {"nodes": sum(entities.values()), "containers": entities}
+
+    async def graph_query(
+        self, text: str, limit: int = 10,
+    ) -> List[Dict[str, Any]]:
+        """Uniform cross-graph query.
+
+        Adapts this graph's search() or query() method by parameter name.
+        Subclasses may override for a better native path.
+        """
+        import asyncio
+        import inspect
+
+        fn = getattr(self, "search", None) or getattr(self, "query", None)
+        if not callable(fn):
+            return []
+
+        try:
+            sig = inspect.signature(fn)
+        except (TypeError, ValueError):
+            return []
+
+        # Bind by parameter name (not position)
+        kwargs = {}
+        for p in sig.parameters.values():
+            if p.name in ("query", "query_str", "text", "prompt", "search"):
+                kwargs[p.name] = text
+            elif p.name in ("limit", "max_results", "top_k", "k"):
+                kwargs[p.name] = limit
+
+        # Call sync or async
+        try:
+            if asyncio.iscoroutinefunction(fn):
+                return await fn(**kwargs)
+            else:
+                loop = asyncio.get_event_loop()
+                return await loop.run_in_executor(None, fn, **kwargs)
+        except TypeError:
+            # Fall back to no kwargs if binding failed
+            try:
+                if asyncio.iscoroutinefunction(fn):
+                    return await fn(text, limit)
+                else:
+                    loop = asyncio.get_event_loop()
+                    return await loop.run_in_executor(None, lambda: fn(text, limit))
+            except Exception:
+                return []
+        except Exception:
+            return []

awgraph/degradation.py

@@ -0,0 +1,48 @@
+"""DegradationRegistry shim for awgraph.
+
+Replaces lib.core.DegradationRegistry with a no-op that tracks optional
+dependencies gracefully.
+"""
+
+from enum import Enum
+from typing import Any, Optional
+
+
+class SubsystemTier(Enum):
+    """Severity of a subsystem failure."""
+    CORE = "core"
+    CRITICAL = "critical"
+    COGNITIVE = "cognitive"
+    AUXILIARY = "auxiliary"
+
+
+class DegradationRegistry:
+    """Track optional dependency availability.
+
+    No-op in the public package; just silently tracks OK/failed states.
+    """
+
+    def __init__(self):
+        self._ok: dict[str, Any] = {}
+        self._failed: dict[str, Any] = {}
+
+    def register_ok(self, name: str, module: str, tier: SubsystemTier) -> None:
+        """Mark a subsystem as OK."""
+        self._ok[name] = {"module": module, "tier": tier}
+
+    def register_failed(
+        self, name: str, module: str, error: Exception, tier: SubsystemTier,
+    ) -> None:
+        """Mark a subsystem as failed."""
+        self._failed[name] = {"module": module, "error": str(error), "tier": tier}
+
+
+_GLOBAL_REGISTRY: Optional[DegradationRegistry] = None
+
+
+def get_registry() -> DegradationRegistry:
+    """Get the global degradation registry."""
+    global _GLOBAL_REGISTRY
+    if _GLOBAL_REGISTRY is None:
+        _GLOBAL_REGISTRY = DegradationRegistry()
+    return _GLOBAL_REGISTRY