databao-context-engine 0.1.3__py3-none-any.whl → 0.1.4.dev1__py3-none-any.whl

databao_context_engine/build_sources/plugin_execution.py

@@ -2,13 +2,13 @@ from dataclasses import dataclass
 from datetime import datetime
 from typing import Any, cast
 
+from databao_context_engine.datasources.types import DatasourceId, PreparedConfig, PreparedDatasource
 from databao_context_engine.pluginlib.build_plugin import (
     BuildDatasourcePlugin,
     BuildFilePlugin,
     BuildPlugin,
 )
 from databao_context_engine.pluginlib.plugin_utils import execute_datasource_plugin, execute_file_plugin
-from databao_context_engine.datasources.types import PreparedConfig, PreparedDatasource, DatasourceId
 
 
 @dataclass()

databao_context_engine/cli/add_datasource_config.py

@@ -6,11 +6,11 @@ from typing import Any
 import click
 
 from databao_context_engine import (
+    ConfigPropertyDefinition,
     DatabaoContextPluginLoader,
     DatabaoContextProjectManager,
     DatasourceId,
     DatasourceType,
-    ConfigPropertyDefinition,
 )
 from databao_context_engine.pluginlib.config import ConfigUnionPropertyDefinition
 

databao_context_engine/cli/info.py

@@ -15,6 +15,7 @@ def _generate_info_string(command_info: DceInfo) -> str:
     info_lines = []
     info_lines.append(f"Databao context engine version: {command_info.version}")
     info_lines.append(f"Databao context engine storage dir: {command_info.dce_path}")
+    info_lines.append(f"Databao context engine plugins: {command_info.plugin_ids}")
 
     info_lines.append("")
 

databao_context_engine/introspection/property_extract.py

@@ -1,15 +1,15 @@
 import types
 from dataclasses import MISSING, fields, is_dataclass
-from typing import Annotated, Any, ForwardRef, Union, get_origin, get_type_hints, get_args
+from typing import Annotated, Any, ForwardRef, Union, get_args, get_origin, get_type_hints
 
-from pydantic import BaseModel, _internal
+from pydantic import BaseModel
 from pydantic_core import PydanticUndefinedType
 
 from databao_context_engine.pluginlib.config import (
     ConfigPropertyAnnotation,
     ConfigPropertyDefinition,
-    ConfigUnionPropertyDefinition,
     ConfigSinglePropertyDefinition,
+    ConfigUnionPropertyDefinition,
 )
 
 
@@ -59,18 +59,14 @@ def _get_property_list_from_dataclass(parent_type: type) -> list[ConfigPropertyD
         raise ValueError(f"{parent_type} is not a dataclass")
 
     dataclass_fields = fields(parent_type)
+    type_hints = get_type_hints(parent_type, include_extras=True)
 
     result = []
     for field in dataclass_fields:
         has_field_default = field.default is not None and field.default != MISSING
 
-        if isinstance(field.type, str):
-            try:
-                property_type = _evaluate_type_string(field.type)
-            except Exception:
-                continue
-        else:
-            property_type = field.type
+        # Use the type hints if the field type wasn't resolved (aka. if it is a ForwardRef or a str)
+        property_type = type_hints[field.name] if isinstance(field.type, ForwardRef | str) else field.type
 
         property_for_field = _create_property(
             property_type=property_type,
@@ -89,6 +85,10 @@ def _get_property_list_from_pydantic_base_model(parent_type: type):
     if not issubclass(parent_type, BaseModel):
         raise ValueError(f"{parent_type} is not a Pydantic BaseModel")
 
+    if any(isinstance(field.annotation, ForwardRef) for field in parent_type.model_fields.values()):
+        # If any field's future type wasn't resolved yet, we rebuild the model to resolve them
+        parent_type.model_rebuild(force=True)
+
     pydantic_fields = parent_type.model_fields
     result = []
 
@@ -214,18 +214,3 @@ def compute_default_value(
         return str(property_default)
 
     return None
-
-
-def _evaluate_type_string(property_type: str) -> type:
-    try:
-        # Using a pydantic internal function for this, to avoid having to implement type evaluation manually...
-        return _internal._typing_extra.eval_type(property_type)
-    except Exception as initial_error:
-        try:
-            # Try to convert it ourselves if Pydantic didn't work
-            return ForwardRef(property_type)._evaluate(  # type: ignore[return-value]
-                globalns=globals(), localns=locals(), recursive_guard=frozenset()
-            )
-        except Exception as e:
-            # Ignore if we didn't manage to convert the str to a type
-            raise e from initial_error

databao_context_engine/mcp/mcp_server.py

@@ -1,5 +1,6 @@
 import logging
 from contextlib import asynccontextmanager
+from datetime import date
 from pathlib import Path
 from typing import Literal
 
@@ -7,8 +8,6 @@ from mcp.server import FastMCP
 from mcp.types import ToolAnnotations
 
 from databao_context_engine import DatabaoContextEngine
-from databao_context_engine.mcp.all_results_tool import run_all_results_tool
-from databao_context_engine.mcp.retrieve_tool import run_retrieve_tool
 
 logger = logging.getLogger(__name__)
 
@@ -41,19 +40,23 @@ class McpServer:
             annotations=ToolAnnotations(readOnlyHint=True, idempotentHint=True, openWorldHint=False),
         )
         def all_results_tool():
-            return run_all_results_tool(self._databao_context_engine)
+            return self._databao_context_engine.get_all_contexts_formatted()
 
         @mcp.tool(
             description="Retrieve the context built from various resources, including databases, dbt tools, plain and structured files, to retrieve relevant information",
             annotations=ToolAnnotations(readOnlyHint=True, idempotentHint=True, openWorldHint=False),
         )
         def retrieve_tool(text: str, limit: int | None):
-            return run_retrieve_tool(
-                databao_context_engine=self._databao_context_engine,
-                text=text,
-                limit=limit or 50,
+            retrieve_results = self._databao_context_engine.search_context(
+                retrieve_text=text, limit=limit, export_to_file=False
             )
 
+            display_results = [context_search_result.context_result for context_search_result in retrieve_results]
+
+            display_results.append(f"\nToday's date is {date.today()}")
+
+            return "\n".join(display_results)
+
         return mcp
 
     def run(self, transport: McpTransport):

databao_context_engine/plugins/base_db_plugin.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, TypeVar
+from typing import Annotated, Any, TypeVar
 
 from pydantic import BaseModel, ConfigDict, Field
 
@@ -8,6 +8,7 @@ from databao_context_engine.pluginlib.build_plugin import (
     BuildDatasourcePlugin,
     EmbeddableChunk,
 )
+from databao_context_engine.pluginlib.config import ConfigPropertyAnnotation
 from databao_context_engine.plugins.databases.base_introspector import BaseIntrospector
 from databao_context_engine.plugins.databases.database_chunker import build_database_chunks
 from databao_context_engine.plugins.databases.introspection_scope import IntrospectionScope
@@ -17,7 +18,9 @@ class BaseDatabaseConfigFile(BaseModel):
     model_config = ConfigDict(populate_by_name=True)
     name: str | None = Field(default=None)
     type: str
-    introspection_scope: IntrospectionScope | None = Field(default=None, alias="introspection-scope")
+    introspection_scope: Annotated[
+        IntrospectionScope | None, ConfigPropertyAnnotation(ignored_for_config_wizard=True)
+    ] = Field(default=None, alias="introspection-scope")
 
 
 T = TypeVar("T", bound=BaseDatabaseConfigFile)

databao_context_engine/plugins/databases/athena_introspector.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Annotated
+from typing import Annotated, Any
 
 from pyathena import connect
 from pyathena.cursor import DictCursor

databao_context_engine/plugins/databases/clickhouse_introspector.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Annotated
+from typing import Annotated, Any
 
 import clickhouse_connect
 from pydantic import BaseModel, Field

databao_context_engine/plugins/databases/duckdb_introspector.py

@@ -7,6 +7,7 @@ from databao_context_engine.plugins.base_db_plugin import BaseDatabaseConfigFile
 from databao_context_engine.plugins.databases.base_introspector import BaseIntrospector, SQLQuery
 from databao_context_engine.plugins.databases.databases_types import DatabaseSchema
 from databao_context_engine.plugins.databases.introspection_model_builder import IntrospectionModelBuilder
+from databao_context_engine.plugins.duckdb_tools import fetchall_dicts
 
 
 class DuckDBConfigFile(BaseDatabaseConfigFile):
@@ -319,7 +320,4 @@ class DuckDBIntrospector(BaseIntrospector[DuckDBConfigFile]):
 
     def _fetchall_dicts(self, connection, sql: str, params) -> list[dict]:
         cur = connection.cursor()
-        cur.execute(sql, params or [])
-        columns = [desc[0].lower() for desc in cur.description] if cur.description else []
-        rows = cur.fetchall()
-        return [dict(zip(columns, row)) for row in rows]
+        return fetchall_dicts(cur, sql, params)

databao_context_engine/plugins/databases/introspection_model_builder.py

@@ -5,13 +5,13 @@ from databao_context_engine.plugins.databases.databases_types import (
     CheckConstraint,
     DatabaseColumn,
     DatabasePartitionInfo,
+    DatabaseSchema,
     DatabaseTable,
     DatasetKind,
     ForeignKey,
     ForeignKeyColumnMap,
     Index,
     KeyConstraint,
-    DatabaseSchema,
 )
 
 

databao_context_engine/plugins/databases/mssql_introspector.py

@@ -1,9 +1,9 @@
 from __future__ import annotations
 
-from typing import Any, Mapping, Annotated
+from typing import Annotated, Any, Mapping
 
 from mssql_python import connect  # type: ignore[import-untyped]
-from pydantic import Field, BaseModel
+from pydantic import BaseModel, Field
 
 from databao_context_engine.pluginlib.config import ConfigPropertyAnnotation
 from databao_context_engine.plugins.base_db_plugin import BaseDatabaseConfigFile

databao_context_engine/plugins/databases/mysql_introspector.py

@@ -1,4 +1,4 @@
-from typing import Any, Annotated
+from typing import Annotated, Any
 
 import pymysql
 from pydantic import BaseModel, Field

databao_context_engine/plugins/databases/snowflake_introspector.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List, Annotated
+from typing import Annotated, Any, Dict, List
 
 import snowflake.connector
 from pydantic import BaseModel, Field

databao_context_engine/plugins/duckdb_tools.py

@@ -0,0 +1,18 @@
+from typing import Any
+
+from databao_context_engine.pluginlib.config import DuckDBSecret
+
+
+def generate_create_secret_sql(secret_name, duckdb_secret: DuckDBSecret) -> str:
+    parameters = [("type", duckdb_secret.type)] + list(duckdb_secret.properties.items())
+    return f"""CREATE SECRET {secret_name} (
+    {", ".join([f"{k} '{v}'" for (k, v) in parameters])}
+);
+"""
+
+
+def fetchall_dicts(cur, sql: str, params=None) -> list[dict[str, Any]]:
+    cur.execute(sql, params or [])
+    columns = [desc[0].lower() for desc in cur.description] if cur.description else []
+    rows = cur.fetchall()
+    return [dict(zip(columns, row)) for row in rows]

databao_context_engine/plugins/resources/parquet_introspector.py

@@ -10,6 +10,7 @@ from duckdb import DuckDBPyConnection
 from pydantic import BaseModel, Field
 
 from databao_context_engine.pluginlib.config import DuckDBSecret
+from databao_context_engine.plugins.duckdb_tools import fetchall_dicts, generate_create_secret_sql
 
 parquet_type = "resources/parquet"
 
@@ -49,14 +50,6 @@ class ParquetIntrospectionResult:
     files: list[ParquetFile]
 
 
-def generate_create_secret_sql(secret_name, duckdb_secret: DuckDBSecret) -> str:
-    parameters = [("type", duckdb_secret.type)] + list(duckdb_secret.properties.items())
-    return f"""CREATE SECRET {secret_name} (
-    {", ".join([f"{k} {v}" for (k, v) in parameters])}
-);
-"""
-
-
 @contextlib.contextmanager
 def _create_secret(conn: DuckDBPyConnection, duckdb_secret: DuckDBSecret):
     secret_name = duckdb_secret.name or "gen_secret_" + str(uuid.uuid4()).replace("-", "_")
@@ -97,10 +90,9 @@ class ParquetIntrospector:
         with self._connect(file_config) as conn:
             with conn.cursor() as cur:
                 resolved_url = _resolve_url(file_config)
-                cur.execute(f"SELECT * FROM parquet_file_metadata('{resolved_url}') LIMIT 1")
-                columns = [desc[0].lower() for desc in cur.description] if cur.description else []
-                rows = cur.fetchall()
-                parquet_file_metadata = [dict(zip(columns, row)) for row in rows]
+                parquet_file_metadata = fetchall_dicts(
+                    cur, f"SELECT * FROM parquet_file_metadata('{resolved_url}') LIMIT 1"
+                )
                 if not parquet_file_metadata:
                     raise ValueError(f"No parquet files found by url {resolved_url}")
                 if not parquet_file_metadata or not parquet_file_metadata[0]["file_name"]:
@@ -110,10 +102,7 @@ class ParquetIntrospector:
         with self._connect(file_config) as conn:
             with conn.cursor() as cur:
                 resolved_url = _resolve_url(file_config)
-                cur.execute(f"SELECT * from parquet_metadata('{resolved_url}')")
-                cols = [desc[0].lower() for desc in cur.description] if cur.description else []
-                rows = cur.fetchall()
-                file_metas = [dict(zip(cols, row)) for row in rows]
+                file_metas = fetchall_dicts(cur, f"SELECT * from parquet_metadata('{resolved_url}')")
 
                 columns_per_file: dict[str, dict[int, ParquetColumn]] = defaultdict(defaultdict)
                 for file_meta in file_metas:

databao_context_engine/project/info.py

@@ -3,6 +3,7 @@ from importlib.metadata import version
 from pathlib import Path
 from uuid import UUID
 
+from databao_context_engine.plugins.plugin_loader import load_plugins
 from databao_context_engine.project.layout import validate_project_dir
 from databao_context_engine.system.properties import get_dce_path
 
@@ -34,6 +35,7 @@ class DceInfo:
 
     version: str
     dce_path: Path
+    plugin_ids: list[str]
 
     project_info: DceProjectInfo
 
@@ -50,6 +52,7 @@ def get_databao_context_engine_info(project_dir: Path) -> DceInfo:
     return DceInfo(
         version=get_dce_version(),
         dce_path=get_dce_path(),
+        plugin_ids=[plugin.id for plugin in load_plugins().values()],
         project_info=_get_project_info(project_dir),
     )
 

{databao_context_engine-0.1.3.dist-info → databao_context_engine-0.1.4.dev1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.3
 Name: databao-context-engine
-Version: 0.1.3
-Summary: Add your description here
+Version: 0.1.4.dev1
+Summary: Semantic context for your LLMs — generated automatically
 Requires-Dist: click>=8.3.0
 Requires-Dist: duckdb>=1.4.3
 Requires-Dist: pyyaml>=6.0.3

{databao_context_engine-0.1.3.dist-info → databao_context_engine-0.1.4.dev1.dist-info}/RECORD

@@ -4,12 +4,12 @@ databao_context_engine/build_sources/build_runner.py,sha256=YXU-EJoJ5nzGePrvWA8n
 databao_context_engine/build_sources/build_service.py,sha256=eywJzpvZUVrj-JYxXii_kxhTBzVUT0xfbNtdAFcpGWQ,1624
 databao_context_engine/build_sources/build_wiring.py,sha256=7Jw78gCy57Zy0e5nUnAqQmPdTkW6tztlHnQmRJ1cNkE,3192
 databao_context_engine/build_sources/export_results.py,sha256=E3HP4uaQKYDrmAJKFWwDT190j7ZKfO791_pjKebTMQM,1773
-databao_context_engine/build_sources/plugin_execution.py,sha256=dcnnZN-JQxmpHMCVdA6pUtbJwfJRHNmaWnu1f0ySL6M,2211
+databao_context_engine/build_sources/plugin_execution.py,sha256=Icl2oJeyL98aD8m0unmnRdtCOtaZE6iRGtmCUFA_eYk,2211
 databao_context_engine/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-databao_context_engine/cli/add_datasource_config.py,sha256=ukgore2ZG1Gen2A8LCoweOxuNhcVonhT3p_nYbtyAow,6371
+databao_context_engine/cli/add_datasource_config.py,sha256=ESAIkLb_g2Bg-WjorZ6JFSwxh23SY-g1T699_fxip54,6371
 databao_context_engine/cli/commands.py,sha256=ntqoThbplgrHPtDy5kKna1Cz_iu8UKJQG6q7G6UmsCs,7106
 databao_context_engine/cli/datasources.py,sha256=kHbXhEnDVaYXTRAE3WeaDGohH1Y9cwyYjjcycQ8eh1U,2376
-databao_context_engine/cli/info.py,sha256=xX9io82Nzl-k7Xcds8WLTGzzBYorEYh10TlFJvXquyg,1119
+databao_context_engine/cli/info.py,sha256=2iegfvRAxMae3KkWMDu-BOnljeJMVQ5u8P5k_8nOKpc,1203
 databao_context_engine/config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databao_context_engine/config/log_config.yaml,sha256=hH2NPumufcyig8L-7Z_DbLQ88_G2N6CAEl5fvHafeAw,360
 databao_context_engine/config/logging.py,sha256=66k7x3ftEdHx-CKqbOss4pqZfGn4nMFv7AzVPs9aVnY,1525
@@ -26,7 +26,7 @@ databao_context_engine/event_journal/writer.py,sha256=abkl2SPHYzO8XAUsjbtrcXm6WQ
 databao_context_engine/generate_configs_schemas.py,sha256=OBX1Motg_d-fmZCUnDoF-3c8dYbDcTOJmhD7F0WJHH8,3017
 databao_context_engine/init_project.py,sha256=enLvYOd5MpGYVdWly7DuLMzPLUEE6130bEuxh0YZft8,1496
 databao_context_engine/introspection/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-databao_context_engine/introspection/property_extract.py,sha256=61cTxGBUgb9px-hA-S-e5Ex68GjNoNoN3HgwuaGK6ss,8223
+databao_context_engine/introspection/property_extract.py,sha256=m7DL8SWNIipi8uE9q4k3TDvlga9YeKyKh85CvFt_fqI,7816
 databao_context_engine/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databao_context_engine/llm/config.py,sha256=geIygR-9maeKbvmcO5J3iNZXKS4LZc7zpKYyaH8vCoI,463
 databao_context_engine/llm/descriptions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -42,10 +42,8 @@ databao_context_engine/llm/runtime.py,sha256=jf_sPDF4s_fChFj2enLcEIBXIWjSjaYNh-l
 databao_context_engine/llm/service.py,sha256=rM_DrF2F7NylxBhtfVGporUcv91eR_nF1mF--Zxh2tc,5978
 databao_context_engine/main.py,sha256=3gWwEc9GZJUFANJtW9Fx4smVzib_sDtFkBGPILjisaI,350
 databao_context_engine/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-databao_context_engine/mcp/all_results_tool.py,sha256=rwd19nhydLtjxaNaYEKQuJkQwuEID0o_1YdwvvyIhls,200
 databao_context_engine/mcp/mcp_runner.py,sha256=bnB2a-_P1zOlDBqpyFIq4n2hWkfwqb6G7dHb3-3Q2I0,468
-databao_context_engine/mcp/mcp_server.py,sha256=p8FZxqa43ZzWuz8EYJn70RLE0a8-eoPH7Pb2QJ7mmiI,2133
-databao_context_engine/mcp/retrieve_tool.py,sha256=U5K5okX7DyrKQ_-1nZ6YOP_oUjIBLEv7JFjz3f-eJMw,611
+databao_context_engine/mcp/mcp_server.py,sha256=fXkl-LSXjWQwYoranWgBUh5qrIcTv3ZOJvKfbnf0HwM,2231
 databao_context_engine/plugin_loader.py,sha256=2vq7rCNQuxQvnGqF758Ke2IzCO-11dLD2lH8BKP1b68,4829
 databao_context_engine/pluginlib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databao_context_engine/pluginlib/build_plugin.py,sha256=v90JU4JRqYb4BbMvx833lPVafBUOE7Bhpt8SOWdBrZU,4105
@@ -53,23 +51,24 @@ databao_context_engine/pluginlib/config.py,sha256=fAu1AuxSFI6LOoAyvwuFE5p11bOsHp
 databao_context_engine/pluginlib/plugin_utils.py,sha256=f76ksdMUG3Wy4KOytUYxLMrq9iOHytu3QF8wc5NzzJE,2471
 databao_context_engine/plugins/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databao_context_engine/plugins/athena_db_plugin.py,sha256=53Dui_V7MunJ3jmyYEAmkZofuHtQP-Ic3bNhsxTwAyc,456
-databao_context_engine/plugins/base_db_plugin.py,sha256=uuzKSEtszKnxL-cBg5ya6TxQHq8oqlIicUbZlkoyxlI,1562
+databao_context_engine/plugins/base_db_plugin.py,sha256=NkDR7aHqskoBwtOJbfiMdn_nWNe94lbq_ZbciiDbfiI,1733
 databao_context_engine/plugins/clickhouse_db_plugin.py,sha256=4J1CNoeWPRvgmgkww7CQALxTP5D9eSLl1TpDRgwHBew,509
 databao_context_engine/plugins/databases/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-databao_context_engine/plugins/databases/athena_introspector.py,sha256=O8CfPzj5y0gCLBAjte8bvo7F9c2PGH6ldP6uq8_0Wt4,5702
+databao_context_engine/plugins/databases/athena_introspector.py,sha256=--F8sbtPdcCUtDfMzx85_zH0sBlArJigM6Aa1mmn00w,5702
 databao_context_engine/plugins/databases/base_introspector.py,sha256=aXMWJ13LdYs1pSaGIhtA8zefe-oXlmzM6X_Ur8ZbMok,5760
-databao_context_engine/plugins/databases/clickhouse_introspector.py,sha256=QfXLaA6gr0pdSLiaC3Mri_JohhRGgjw1rCJxLsHJOqk,6353
+databao_context_engine/plugins/databases/clickhouse_introspector.py,sha256=6EyV46QR9UhcClnmBNxRzxgShvedbOo58p77hPFcK2s,6353
 databao_context_engine/plugins/databases/database_chunker.py,sha256=dn9Mb9Sw4f8IwBe9RtukQGT0yyKLQt5i4tjvsrFh1ew,2100
 databao_context_engine/plugins/databases/databases_types.py,sha256=kYZEaFa_cnROg_n1k7tIonoC2VunLBe5SQR47GOt_8k,2579
-databao_context_engine/plugins/databases/duckdb_introspector.py,sha256=ymcvEM6zodZBY92xjamyZHJBrA2Jk1sFTZetqxR80Pg,11430
-databao_context_engine/plugins/databases/introspection_model_builder.py,sha256=nfQLKVbwaWY12lnWrpCoVY2rolW263IwvWg7NkLjtk0,10350
+databao_context_engine/plugins/databases/duckdb_introspector.py,sha256=temeOVxUY46vOgHZ9g-ra0n5thCEthvGTD7g3dZJkBk,11332
+databao_context_engine/plugins/databases/introspection_model_builder.py,sha256=-i7e4jqwdcsu3NnS-LHVSH_DrG3riN09KX3SDHAGF0c,10350
 databao_context_engine/plugins/databases/introspection_scope.py,sha256=0BZAAmVTiHifOzeuQkCsGKrGRnZCPHRCLqam5nQqr88,2225
 databao_context_engine/plugins/databases/introspection_scope_matcher.py,sha256=gseF4j-SD-7SSPMDlDqHUOTI3tPSaOGKJgp42WDVlsY,3605
-databao_context_engine/plugins/databases/mssql_introspector.py,sha256=Pnu9LpEXZn7qUH0ANHeijynamX6dLJveHrK0tOy1PII,18793
-databao_context_engine/plugins/databases/mysql_introspector.py,sha256=IoiAjzUuJwkjAXPw-X7LBRfKoY3Y2WHrHzEeXNfdF9A,14504
+databao_context_engine/plugins/databases/mssql_introspector.py,sha256=mw6z3YUKQ5hzFFWpNhx5NYrOG6tfQIclWESvdcUhIoY,18793
+databao_context_engine/plugins/databases/mysql_introspector.py,sha256=Y77snyZkTT5elL7bsbzjUeffugLl_v1NB4yMQFRbCso,14504
 databao_context_engine/plugins/databases/postgresql_introspector.py,sha256=tHsQNWJLHcH4GUYwmXSf5mY6XIMhvtpUbGgwmWfljpw,17489
-databao_context_engine/plugins/databases/snowflake_introspector.py,sha256=6AYsJNfhT6rAJYBH7A-_ukjtLxi-0p2Dsz9b6cda4U4,12005
+databao_context_engine/plugins/databases/snowflake_introspector.py,sha256=K3v25O-jMybM9Iey5Wqub5oGLjcHsSaKDu4VXWMzu70,12005
 databao_context_engine/plugins/duckdb_db_plugin.py,sha256=uumf9kXz6U-6QbX0i1ChHHxxvdq5_di3gTNpVDUCYdg,451
+databao_context_engine/plugins/duckdb_tools.py,sha256=46rctnTxDPAhHtaiTp1DxMuuDuRKrtKWJFSSM2w7uUU,645
 databao_context_engine/plugins/mssql_db_plugin.py,sha256=YVor2QMDuL6XOLT8RFPOgs1Uhj1ZVYoFP5DApDJFPaE,446
 databao_context_engine/plugins/mysql_db_plugin.py,sha256=wy8HTiYuE8GTxfoc5gdrinEYv7NWHGrslzaAzruQecY,446
 databao_context_engine/plugins/parquet_plugin.py,sha256=qlpzQmiEwpb8Jma50HoIvljGV5ZiWkZU7GSiGV7M6kc,1181
@@ -77,11 +76,11 @@ databao_context_engine/plugins/plugin_loader.py,sha256=Nx_OYGf9t3XKb_JV4Om9fCOuL
 databao_context_engine/plugins/postgresql_db_plugin.py,sha256=PwLErcbqixsgz4Tqoift0uyerUqP_OaRzMaMAn0TE5A,486
 databao_context_engine/plugins/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databao_context_engine/plugins/resources/parquet_chunker.py,sha256=R9WCOBqpKRTVN6t5eeOm_mmnKBOxvjIiQ9zTc8vnUb4,848
-databao_context_engine/plugins/resources/parquet_introspector.py,sha256=_nB85_cSeFAHbnYlGPP_apRFkUQ6j24Efv-KNtOF2tM,6149
+databao_context_engine/plugins/resources/parquet_introspector.py,sha256=dnf_rLQYd9RWIevl64_8ddhXO5AFQhcuOWQiaRCOeYM,5639
 databao_context_engine/plugins/snowflake_db_plugin.py,sha256=ApiVE212DU9F4Zj1HPof6JQ66n4LUeYRRwhlTWRCxek,486
 databao_context_engine/plugins/unstructured_files_plugin.py,sha256=TWV3fluusizul2YEN-YZ-K-C3s5W-5B8353-bjP53L8,2424
 databao_context_engine/project/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-databao_context_engine/project/info.py,sha256=XcDB2tKlsM6PCKapXEB3fLIHPSWUR3XlNjnKWLyJ92s,2313
+databao_context_engine/project/info.py,sha256=3chVSZ4pV2EsYJSeVfOYTK746POlJImciIBOvrQuYbc,2479
 databao_context_engine/project/init_project.py,sha256=QiLalH--BMQGB-c2WBbPAUKBinDbO0cJkmilUfqPAuI,3970
 databao_context_engine/project/layout.py,sha256=7ThzqnFejMogBV8c6Mt-Md4GO-kdzYNrmhhBTalYvps,4295
 databao_context_engine/project/project_config.py,sha256=rI-Wkll7lca7RlYIaFScs5CIKIZ8uujtKRxSjibGIt8,1132
@@ -120,7 +119,7 @@ databao_context_engine/system/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 databao_context_engine/system/properties.py,sha256=mQ7-_PZeYSESYn1cMUQ0IK7rJEnbhc7t4WesFjAgo-Q,429
 databao_context_engine/templating/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 databao_context_engine/templating/renderer.py,sha256=W2-0IGStAp6oxANmsKs_Z-UoIR6Gt_c4ILYFa3Hruo4,662
-databao_context_engine-0.1.3.dist-info/WHEEL,sha256=eycQt0QpYmJMLKpE3X9iDk8R04v2ZF0x82ogq-zP6bQ,79
-databao_context_engine-0.1.3.dist-info/entry_points.txt,sha256=5EeQJ1W8zEFh4HuF1bs2zBeoP408oiwuM9UrkJiurgI,138
-databao_context_engine-0.1.3.dist-info/METADATA,sha256=81FImIbcFo2PNXdEHjECQzrL7wCMTBpUc51Px6Fv7Go,3329
-databao_context_engine-0.1.3.dist-info/RECORD,,
+databao_context_engine-0.1.4.dev1.dist-info/WHEEL,sha256=5w2T7AS2mz1-rW9CNagNYWRCaB0iQqBMYLwKdlgiR4Q,78
+databao_context_engine-0.1.4.dev1.dist-info/entry_points.txt,sha256=5EeQJ1W8zEFh4HuF1bs2zBeoP408oiwuM9UrkJiurgI,138
+databao_context_engine-0.1.4.dev1.dist-info/METADATA,sha256=43INtHVdc-AdDCHzRA-bHxGP66x_-kaXbKi1ZSapp3M,3367
+databao_context_engine-0.1.4.dev1.dist-info/RECORD,,

{databao_context_engine-0.1.3.dist-info → databao_context_engine-0.1.4.dev1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.24
+Generator: uv 0.9.7
 Root-Is-Purelib: true
 Tag: py3-none-any

databao_context_engine/mcp/all_results_tool.py

@@ -1,5 +0,0 @@
-from databao_context_engine import DatabaoContextEngine
-
-
-def run_all_results_tool(databao_context_engine: DatabaoContextEngine) -> str:
-    return databao_context_engine.get_all_contexts_formatted()

databao_context_engine/mcp/retrieve_tool.py

@@ -1,14 +0,0 @@
-import datetime
-
-from databao_context_engine import DatabaoContextEngine
-
-
-def run_retrieve_tool(*, databao_context_engine: DatabaoContextEngine, text: str, limit: int | None = None) -> str:
-    """Execute the retrieve flow for MCP and return the matching display texts."""
-    retrieve_results = databao_context_engine.search_context(retrieve_text=text, limit=limit, export_to_file=False)
-
-    display_results = [context_search_result.context_result for context_search_result in retrieve_results]
-
-    display_results.append(f"\nToday's date is {datetime.date.today()}")
-
-    return "\n".join(display_results)