krail 0.2.1__py3-none-any.whl

krail-0.2.1.dist-info/METADATA

@@ -0,0 +1,74 @@
+Metadata-Version: 2.4
+Name: krail
+Version: 0.2.1
+Summary: KRAIL client for local projects and the local API runtime
+Author: AkeBoss Tech
+License-Expression: MIT
+Project-URL: Homepage, https://github.com/AkeBoss-tech/knowledge
+Project-URL: Repository, https://github.com/AkeBoss-tech/knowledge
+Project-URL: Issues, https://github.com/AkeBoss-tech/knowledge/issues
+Keywords: knowledge,research,agents,local-first,mcp
+Classifier: Development Status :: 3 - Alpha
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Scientific/Engineering :: Information Analysis
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: httpx>=0.27
+Requires-Dist: pydantic>=2
+Requires-Dist: pyyaml>=6
+Provides-Extra: analysis
+Requires-Dist: numpy; extra == "analysis"
+Requires-Dist: pandas; extra == "analysis"
+Requires-Dist: statsmodels; extra == "analysis"
+Requires-Dist: matplotlib; extra == "analysis"
+Provides-Extra: local
+Requires-Dist: owlready2; extra == "local"
+Requires-Dist: duckdb; extra == "local"
+Provides-Extra: embeddings
+Requires-Dist: sentence-transformers>=2.7; extra == "embeddings"
+
+# krail
+
+KRAIL client supporting local project mode and the local FastAPI runtime.
+
+Install from PyPI:
+
+```bash
+pip install krail
+```
+
+The distribution is named `krail`; the Python import namespace remains `rail`.
+
+## Usage Examples
+
+```python
+import rail
+
+# API mode
+project = rail.connect("nj-economics")
+
+# DataFrame queries
+df = project.query("SELECT county_name, unemployment_rate FROM County ORDER BY unemployment_rate DESC LIMIT 10")
+
+# Agent research
+answer = project.agent.ask("What counties had unemployment above 10% in 2020?")
+print(answer)
+
+# Streaming agent
+for event in project.agent.ask("Compare Hudson and Bergen County unemployment trends", stream=True):
+    if event["type"] == "text_delta":
+        print(event["text"], end="", flush=True)
+
+# Direct local mode
+project = rail.local("./nj-economics")
+ont = project.ontology()
+counties = ont.individuals("County")
+print(f"Loaded {len(counties)} counties")
+```

krail-0.2.1.dist-info/RECORD

@@ -0,0 +1,26 @@
+rail/__init__.py,sha256=DbCsxwtTQpdMvGj1msSfssBIgTHCd696AMyE6PB6evY,1480
+rail/agent.py,sha256=rXq_LWusbUIrvdhMhk9zxiBYymAxtaoXEGW6HqOsJa4,2589
+rail/bootstrap.py,sha256=MXq_KWY66IrCfj9XjvD62ojca2dfjpgGuUTil0vmecc,49402
+rail/cli.py,sha256=QOFyqQSbouF9m3RlqtRH86bc02ZJyWQ6txYw-Ewzh_s,44619
+rail/client.py,sha256=VRCpz-oaFR6JyIhb_-DPV3a022DlyFqCOXUtfg92nzc,5155
+rail/completion_gate.py,sha256=WqG1h4e690sohU2tptOtiHymiNUTrZyxmWGOIq0BvTU,8940
+rail/exceptions.py,sha256=0SPYgg1fUz0L72miE6S-BCiQ3s0afDJmM3KQTm6mLT0,299
+rail/integrity.py,sha256=3904cIKQUZbGR6-kSUAybnjOdygBUN2EzTIJ84bexN0,210824
+rail/knowledge.py,sha256=I3mL-eUYh6kBgQDNiAg_H3yXz_MC5vgvtP2flVEVk7c,147494
+rail/local.py,sha256=2tikTligGjDcSc7tN8lYPdD6I0a_jn8DuFku1BrealA,28718
+rail/manifest.py,sha256=-S9J5w3zJIFu9DAp13iaWkf8xpjIJbjcqJ-LLunSKJA,18714
+rail/markdown_graph.py,sha256=2zh4O6Ykd6IMn37ZvwgOlc8SrMu5ZRpdIAhDsEhTuhM,22133
+rail/models.py,sha256=uXHhrtO6fU12YvvCHsLEz_9Tt6MELUJ1AF-o9RRucqo,516
+rail/modes.py,sha256=ubrwKWD-zDQF50yYY_y-0yfEWu6JSmiUly2vOqJRfjc,6730
+rail/ontology.py,sha256=2uZJ7Gjg4tDlcG-9nYpp-CZ3DskO9ivoozRWqyxhzgg,2202
+rail/planner_sync.py,sha256=zToU3LpR1p35QlIxd1Rog9KGW6AhnMuBoX9zOtjeWY0,8093
+rail/project.py,sha256=jbdbbgZywJ5F_GfZBsxF6YxCwS3wvOZjW78tjjRev-M,28928
+rail/session_state.py,sha256=dSKd15-V4SdcfRFItkYqiycinUg2uv6eoIBssARB404,1431
+rail/source_dependencies.py,sha256=CqygACp1xL96DPgefD_Ly70qWy3aufBaUrXsPrKmxfA,10377
+rail/vector_store.py,sha256=VjUQIQn5OVWEYi0wDNZLGKyRUzcICFvQiNQ_vBZMgiI,10278
+rail/verification.py,sha256=lA2xX2aYXU_hqz_YAmeDQqeR1DimDomWDM7kMJMxzL8,17126
+krail-0.2.1.dist-info/METADATA,sha256=igN7HxIosLS6KreQlOrp8H0BonVgeO7HTCEpMqKZI-g,2404
+krail-0.2.1.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+krail-0.2.1.dist-info/entry_points.txt,sha256=z0p7tay632AJM1f3z9MB-VzBix56I5-beIsEbLzEMgc,61
+krail-0.2.1.dist-info/top_level.txt,sha256=6R6sqn-85I8YZTzmB9gv7sqckB2QaOTHbu-QLGTWBRE,5
+krail-0.2.1.dist-info/RECORD,,

krail-0.2.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

krail-0.2.1.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+krail = rail.cli:main
+rail = rail.cli:main

krail-0.2.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+rail

rail/__init__.py

@@ -0,0 +1,53 @@
+from rail.project import Project
+from rail.client import CloudClient
+from rail.local import LocalEngine
+from rail.integrity import (
+    ArtifactLineageRecord,
+    AssumptionRecord,
+    ClaimRecord,
+    IntegrityIndexes,
+    ResearchIntegrityRepo,
+    SourceRecord,
+    VerificationRunRecord,
+)
+from rail.manifest import (
+    ContractViolation,
+    ManifestValidationError,
+    RailManifest,
+    boot_validate_project,
+    load_and_validate_manifest,
+    load_manifest,
+    parse_manifest_content,
+)
+from rail.session_state import (
+    ACTIVE_STATUSES,
+    TERMINAL_STATUSES,
+    is_active_status,
+    is_terminal_status,
+    normalize_session_record,
+    normalize_session_status,
+)
+from rail.exceptions import RailError
+
+__version__ = "0.2.1"
+
+def connect(
+    slug: str,
+    api_url: str | None = None,
+    api_key: str | None = None,
+) -> Project:
+    """Connect to a KRAIL project via the local FastAPI runtime."""
+    import os
+    url = api_url or os.environ.get("RAIL_API_URL", "http://localhost:8000/api/v1")
+    key = api_key or os.environ.get("RAIL_API_KEY", "")
+    client = CloudClient(base_url=url, api_key=key)
+    return Project(slug=slug, backend=client)
+
+def local(
+    path: str = ".",
+    engine_path: str | None = None,
+) -> Project:
+    """Load a KRAIL project from a local repo directory."""
+    engine = LocalEngine(project_path=path, engine_path=engine_path)
+    slug = engine.read_rail_yaml().project.slug
+    return Project(slug=slug, backend=engine)

rail/agent.py

@@ -0,0 +1,59 @@
+import json
+import httpx
+from typing import Generator, AsyncGenerator
+
+class AgentClient:
+    def __init__(self, base_url: str, project_slug: str, api_key: str = ""):
+        self.base_url = base_url.rstrip("/")
+        self.project_slug = project_slug
+        self.headers = {"Authorization": f"Bearer {api_key}"} if api_key else {}
+
+    def ask(self, question: str, stream: bool = False, model: str | None = None):
+        """Ask a question. Returns full answer text (stream=False) or event generator (stream=True)."""
+        if stream:
+            return self._stream(question, model)
+        else:
+            return self._blocking(question, model)
+
+    def _blocking(self, question: str, model: str | None) -> str:
+        """Collect all text_delta events and return the full answer."""
+        text = []
+        for event in self._stream(question, model):
+            if event.get("type") == "text_delta":
+                text.append(event.get("text", ""))
+        return "".join(text)
+
+    def _stream(self, question: str, model: str | None) -> Generator[dict, None, None]:
+        """Yield raw SSE event dicts."""
+        url = f"{self.base_url}/agent/chat?project={self.project_slug}"
+        payload = {"message": question, "history": []}
+        if model:
+            payload["model"] = model
+
+        with httpx.Client(timeout=300) as client:
+            with client.stream("POST", url, json=payload, headers=self.headers) as resp:
+                resp.raise_for_status()
+                for line in resp.iter_lines():
+                    if line.startswith("data: "):
+                        try:
+                            event = json.loads(line[6:])
+                            yield event
+                        except json.JSONDecodeError:
+                            pass
+
+    async def ask_async(self, question: str, model: str | None = None) -> AsyncGenerator[dict, None]:
+        """Async streaming version."""
+        url = f"{self.base_url}/agent/chat?project={self.project_slug}"
+        payload = {"message": question, "history": []}
+        if model:
+            payload["model"] = model
+
+        async with httpx.AsyncClient(timeout=300) as client:
+            async with client.stream("POST", url, json=payload, headers=self.headers) as resp:
+                resp.raise_for_status()
+                async for line in resp.aiter_lines():
+                    if line.startswith("data: "):
+                        try:
+                            yield json.loads(line[6:])
+                        except json.JSONDecodeError:
+                            pass