databao-context-engine 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

databao_context_engine/storage/repositories/run_repository.py

@@ -1,157 +0,0 @@
-from datetime import datetime
-from typing import Any, Optional
-
-import duckdb
-
-from databao_context_engine.services.run_name_policy import RunNamePolicy
-from databao_context_engine.storage.models import RunDTO
-
-
-class RunRepository:
-    def __init__(self, conn: duckdb.DuckDBPyConnection, run_name_policy: RunNamePolicy):
-        self._conn = conn
-        self._run_name_policy = run_name_policy
-
-    def create(
-        self, *, project_id: str, dce_version: Optional[str] = None, started_at: datetime | None = None
-    ) -> RunDTO:
-        if started_at is None:
-            started_at = datetime.now()
-        run_name = self._run_name_policy.build(run_started_at=started_at)
-
-        row = self._conn.execute(
-            """
-            INSERT INTO 
-                run (project_id, nemory_version, started_at, run_name)
-            VALUES 
-                (?, ?, ?, ?)
-            RETURNING 
-                *
-            """,
-            [project_id, dce_version, started_at, run_name],
-        ).fetchone()
-        if row is None:
-            raise RuntimeError("Run creation returned no object")
-        return self._row_to_dto(row)
-
-    def get(self, run_id: int) -> Optional[RunDTO]:
-        row = self._conn.execute(
-            """
-            SELECT 
-                *
-            FROM
-                run
-            WHERE
-                run_id = ?
-            """,
-            [run_id],
-        ).fetchone()
-        return self._row_to_dto(row) if row else None
-
-    def get_by_run_name(self, *, project_id: str, run_name: str) -> RunDTO | None:
-        row = self._conn.execute(
-            """
-            SELECT
-                *
-            FROM
-                run
-            WHERE
-                run.project_id = ? AND run_name = ?
-            """,
-            [project_id, run_name],
-        ).fetchone()
-        return self._row_to_dto(row) if row else None
-
-    def get_latest_run_for_project(self, project_id: str) -> RunDTO | None:
-        row = self._conn.execute(
-            """
-            SELECT
-                *
-            FROM
-                run
-            WHERE
-                run.project_id = ?
-            ORDER BY run.started_at DESC
-            LIMIT 1
-            """,
-            [project_id],
-        ).fetchone()
-        return self._row_to_dto(row) if row else None
-
-    def update(
-        self,
-        run_id: int,
-        *,
-        project_id: Optional[str] = None,
-        ended_at: Optional[datetime] = None,
-        dce_version: Optional[str] = None,
-    ) -> Optional[RunDTO]:
-        sets: list[Any] = []
-        params: list[Any] = []
-
-        if project_id is not None:
-            sets.append("project_id = ?")
-            params.append(project_id)
-        if ended_at is not None:
-            sets.append("ended_at = ?")
-            params.append(ended_at)
-        if dce_version is not None:
-            sets.append("nemory_version = ?")
-            params.append(dce_version)
-
-        if not sets:
-            return self.get(run_id)
-
-        params.append(run_id)
-        self._conn.execute(
-            f"""
-                    UPDATE
-                        run
-                    SET 
-                        {", ".join(sets)}
-                    WHERE
-                        run_id = ?
-                    """,
-            params,
-        )
-
-        return self.get(run_id)
-
-    def delete(self, run_id: int) -> int:
-        row = self._conn.execute(
-            """
-            DELETE FROM
-                run
-            WHERE
-                run_id = ?
-            RETURNING
-                run_id
-            """,
-            [run_id],
-        ).fetchone()
-        return 1 if row else 0
-
-    def list(self) -> list[RunDTO]:
-        rows = self._conn.execute(
-            """
-            SELECT
-                *
-            FROM
-                run
-            ORDER BY
-                run_id DESC
-            """
-        ).fetchall()
-        return [self._row_to_dto(r) for r in rows]
-
-    @staticmethod
-    def _row_to_dto(row: tuple) -> RunDTO:
-        run_id, project_id, started_at, ended_at, dce_version, run_name = row
-        return RunDTO(
-            run_id=int(run_id),
-            run_name=run_name,
-            project_id=str(project_id),
-            started_at=started_at,
-            ended_at=ended_at,
-            nemory_version=dce_version,
-        )

databao_context_engine-0.1.2.dist-info/METADATA

@@ -1,187 +0,0 @@
-Metadata-Version: 2.3
-Name: databao-context-engine
-Version: 0.1.2
-Summary: Add your description here
-Requires-Dist: click>=8.3.0
-Requires-Dist: duckdb>=1.4.3
-Requires-Dist: pyyaml>=6.0.3
-Requires-Dist: requests>=2.32.5
-Requires-Dist: pymysql>=1.1.2
-Requires-Dist: clickhouse-connect>=0.10.0
-Requires-Dist: mcp>=1.23.3
-Requires-Dist: pyathena>=3.22.0
-Requires-Dist: snowflake-connector-python>=4.1.0
-Requires-Dist: pydantic>=2.12.4
-Requires-Dist: jinja2>=3.1.6
-Requires-Dist: asyncpg>=0.31.0
-Requires-Dist: asyncio>=4.0.0
-Requires-Dist: asyncpg-stubs>=0.31.1
-Requires-Dist: mssql-python>=1.0.0 ; extra == 'mssql'
-Requires-Python: >=3.12
-Provides-Extra: mssql
-Description-Content-Type: text/markdown
-
-[![official project](https://jb.gg/badges/official.svg)](https://confluence.jetbrains.com/display/ALL/JetBrains+on+GitHub)
-[![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](https://github.com/JetBrains/databao-context-engine/blob/main/LICENSE)
-
-[//]: # ([![PyPI version](https://img.shields.io/pypi/v/databao-context-engine.svg)](https://pypi.org/project/databao-context-engine))
-
-[//]: # ([![Python versions](https://img.shields.io/pypi/pyversions/databao-context-engine.svg)](https://pypi.org/project/databao-context-engine/))
-
-
-<h1 align="center">Databao Context Engine</h1>
-<p align="center">
- <b>Semantic context for your LLMs — generated automatically.</b><br/>
- No more copying schemas. No manual documentation. Just accurate answers.
-</p>
-<p align="center">
- <a href="https://databao.app">Website</a> 
-
-[//]: # (•)
-
-[//]: # ( <a href="#quickstart">Quickstart</a> •)
-
-[//]: # ( <a href="#supported-data-sources">Data Sources</a> •)
-
-[//]: # ( <a href="#contributing">Contributing</a>)
-</p>
-
----
-
-## What is Databao Context Engine?
-
-Databao Context Engine **automatically generates governed semantic context** from your databases, BI tools, documents, and spreadsheets.
-
-Integrate it with any LLM to deliver **accurate, context-aware answers** — without copying schemas or writing documentation by hand.
-
-```
-Your data sources → Context Engine → Unified semantic graph → Any LLM
-```
-
-## Why choose Databao Context Engine?
-
-| Feature                    | What it means for you                                          |
-|----------------------------|----------------------------------------------------------------|
-| **Auto-generated context** | Extracts schemas, relationships, and semantics automatically   |
-| **Runs locally**           | Your data never leaves your environment                        |
-| **MCP integration**        | Works with Claude Desktop, Cursor, and any MCP-compatible tool |
-| **Multiple sources**       | Databases, dbt projects, spreadsheets, documents               |
-| **Built-in benchmarks**    | Measure and improve context quality over time                  |
-| **LLM agnostic**           | OpenAI, Anthropic, Ollama, Gemini — use any model              |
-| **Governed & versioned**   | Track, version, and share context across your team             |
-| **Dynamic or static**      | Serve context via MCP server or export as artifact             |
-
-# Prerequisites
-
-This README assumes you will use `uv` as your package manager.
-
-You can install it following the instructions [here](https://docs.astral.sh/uv/getting-started/installation/)
-
-If you are going to push to the repository, please make sure to install git pre-commit hooks by running
-
-```bash
-  uv run pre-commit install
-```
-
-# How to run?
-
-You can run it with:
-
-```bash
-  uv run dce info
-```
-
-Not providing the `info` subcommand or using the `--help` flag will show the help screen for the command.
-
-## Using the dce command directly
-
-To be able to use the `dce` command directly (without using `uv run` or `python`) there are two options.
-
-### Installing dce locally
-
-For that one needs to:
-
-1. Build the project by running
-
-```bash
-  uv build
-```
-
-2. Installing the project on our machine by running:
-
-```bash
-  uv tool install -e .
-```
-
-This second step will install the `dce` script on your machine and add it into your path.
-
-### Create dce alias using nix
-
-This method will simply create a new shell environment with `dce` alias. For that one needs to install `nix` package
-manager (https://nixos.org/download/). After that one could simply run in the project root
-
-```bash
-$ nix-shell
-```
-
-which is a short version of `$ nix-shell shell.nix`.
-
-Alternatively, one could specify the path to the project repository
-
-```bash
-$ nix-shell {path_to_dce_repository}
-```
-
-After that, you can then directly use:
-
-```bash
-  dce --help
-```
-
-Note: when we actually release our built Python package, users that don't use `uv` will still be able to install the CLI
-by using `pipx install` instead.
-
-# Running Mypy
-
-[mypy](https://mypy.readthedocs.io/en/stable/getting_started.html) has been added to the project for type checking.
-
-You can run it with the following:
-
-```bash
-  uv run mypy src --exclude "test_*" --exclude dist
-```
-
-NB: the above runs type checking on all files within the `src` directory, excluding all test files.
-
-# Running tests
-
-You can run the tests with:
-
-```bash
-  uv run pytest
-```
-
-(there is currently one test succeeding and one test failing in the project)
-
-# Generating JSON Schemas for our plugin's config files
-
-To be able to build a datasource, each plugin requires a yaml config file that describes how to connect to the
-datasource,
-as well as other information needed to customise the plugin.
-
-To document what each config file should look like, we can generate a JSON schema describing the fields allowed in that
-file.
-
-You can generate all JSON schemas for all plugins by running:
-
-```bash
-  uv run generate_configs_schemas
-```
-
-Some options can be provided to the command to choose which plugins to include or exclude from the generation.
-To see the options available, you can refer to the help:
-
-```bash
-  uv run generate_configs_schemas --help
-```
-