databao-context-engine 0.1.4.dev1__tar.gz → 0.1.6__tar.gz

databao_context_engine-0.1.6/PKG-INFO

@@ -0,0 +1,228 @@
+Metadata-Version: 2.3
+Name: databao-context-engine
+Version: 0.1.6
+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
+Requires-Dist: requests>=2.32.5
+Requires-Dist: mcp>=1.23.3
+Requires-Dist: pydantic>=2.12.4
+Requires-Dist: jinja2>=3.1.6
+Requires-Dist: pyathena>=3.25.0 ; extra == 'athena'
+Requires-Dist: clickhouse-connect>=0.10.0 ; extra == 'clickhouse'
+Requires-Dist: mssql-python>=1.0.0 ; extra == 'mssql'
+Requires-Dist: pymysql>=1.1.2 ; extra == 'mysql'
+Requires-Dist: asyncpg>=0.31.0 ; extra == 'postgresql'
+Requires-Dist: snowflake-connector-python>=4.2.0 ; extra == 'snowflake'
+Requires-Python: >=3.12
+Provides-Extra: athena
+Provides-Extra: clickhouse
+Provides-Extra: mssql
+Provides-Extra: mysql
+Provides-Extra: postgresql
+Provides-Extra: snowflake
+Description-Content-Type: text/markdown
+
+[![official project](https://jb.gg/badges/official.svg)](https://github.com/JetBrains#jetbrains-on-github)
+[![PyPI version](https://img.shields.io/pypi/v/databao-context-engine.svg)](https://pypi.org/project/databao-context-engine)
+[![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](https://github.com/JetBrains/databao-context-engine?tab=License-1-ov-file)
+
+[//]: # ([![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 is a CLI tool that **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             |
+
+## Installation
+
+Databao Context Engine is [available on PyPI](https://pypi.org/project/databao-context-engine/) and can be installed with uv, pip, or another package manage.
+
+### Using uv
+
+1. Install Databao Context Engine:
+
+   ```bash
+   uv tool install databao-context-engine
+   ```
+
+1. Add it to your PATH:
+
+   ```bash
+   uv tool update-shell
+   ```
+
+1. Verify the installation:
+
+   ```bash
+   dce --help
+   ```
+
+### Using pip
+
+1. Install Databao Context Engine:
+
+   ```bash
+   pip install databao-context-engine
+   ```
+
+1. Verify the installation:
+
+   ```bash
+   dce --help
+   ```
+
+##  Supported data sources
+
+* <img src="https://cdn.simpleicons.org/postgresql/316192" width="16" height="16" alt=""> PostgreSQL
+* <img src="https://cdn.simpleicons.org/mysql/4479A1" width="16" height="16" alt=""> MySQL
+* <img src="https://cdn.simpleicons.org/sqlite/003B57" width="16" height="16" alt=""> SQLite
+* <img src="https://cdn.simpleicons.org/duckdb/FFF000" width="16" height="16" alt=""> DuckDB
+* <img src="https://cdn.simpleicons.org/dbt/FF694B" width="16" height="16" alt=""> dbt projects
+* 📄 Documents & spreadsheets *(coming soon)*
+
+##  Supported LLMs
+
+| Provider      | Configuration                                |
+|---------------|----------------------------------------------|
+| **Ollama**    | `languageModel: OLLAMA`: runs locally, free  |
+| **OpenAI**    | `languageModel: OPENAI`: requires an API key |
+| **Anthropic** | `languageModel: CLAUDE`: requires an API key |
+| **Google**    | `languageModel: GEMINI`: requires an API key |
+
+## Quickstart
+
+### 1. Create a project
+
+1. Create a new directory for your project and navigate to it:
+
+   ```bash
+   mkdir dce-project && cd dce-project
+   ```
+
+1. Initialize a new project:
+
+   ```bash
+   dce init
+   ```
+
+### 2. Configure data sources
+
+1. When prompted, agree to create a new datasource.
+   You can also use the `dce datasource add` command.
+
+1. Provide the data source type and its name.
+
+1. Open the config file that was created for you in your editor and fill in the connection details.
+
+1. Repeat these steps for all data sources you want to include in your project.
+
+1. If you have data in Markdown or text files,
+   you can add them to the `dce/src/files` directory.
+
+### 3. Build context
+
+1. To build the context, run the following command:
+
+   ```bash
+   dce build
+   ```
+
+### 4. Use Context with Your LLM
+
+**Option A: Dynamic via MCP Server**
+
+Databao Context Engine exposes the context through a local MCP Server, so your agent can access the latest context at runtime.
+
+1. In **Claude Desktop**, **Cursor**, or another MCP-compatible agent, add the following configuration.
+   Replace `dce-project/` with the path to your project directory:
+  
+   ```json 
+   # claude_desktop_config.json, mcp.json, or similar
+   
+   {
+     "mcpServers": {
+       "dce": {
+         "command": "dce mcp",
+         "args": ["--project-dir", "dce-project/"]
+       }
+     }
+   }
+   ```
+
+1. Save the file and restart your agent.
+
+1. Open a new chat, in the chat window, select the `dce` server, and ask questions related to your project context.
+
+**Option B: Static artifact**
+
+Even if you don’t have Claude or Cursor installed on your local machine,
+you can still use the context built by Databao Context Engine by pasting it directly into your chat with an AI assistant.
+
+1. Navigate to `dce-project/output/` and open the directory with the latest run.
+
+1. Attach the `all_results.yaml` file to your chat with the AI assistant or copy and paste its contents into your chat.
+
+##  Contributing
+
+We’d love your help! Here’s how to get involved:
+
+- ⭐ **Star this repo** — it helps others find us!
+- 🐛 **Found a bug?** [Open an issue](https://github.com/JetBrains/databao-context-engine/issues)
+- 💡 **Have an idea?** We’re all ears — create a feature request
+- 👍 **Upvote issues** you care about — helps us prioritize
+- 🔧 **Submit a PR**
+- 📝 **Improve docs** — typos, examples, tutorials — everything helps!
+
+New to open source? No worries! We're friendly and happy to help you get started. 🌱
+
+For more details, see [CONTRIBUTING](CONTRIBUTING.md).
+
+## 📄 License
+
+Apache 2.0 — use it however you want. See the [LICENSE](LICENSE.md) file for details.
+
+---
+
+<p align="center">
+ <b>Like Databao Context Engine?</b> Give us a ⭐ — it means a lot!
+</p>
+
+<p align="center">
+ <a href="https://databao.app">Website</a> •
+ <a href="https://discord.gg/databao">Discord</a>
+</p>

databao_context_engine-0.1.6/README.md

@@ -0,0 +1,202 @@
+[![official project](https://jb.gg/badges/official.svg)](https://github.com/JetBrains#jetbrains-on-github)
+[![PyPI version](https://img.shields.io/pypi/v/databao-context-engine.svg)](https://pypi.org/project/databao-context-engine)
+[![License](https://img.shields.io/badge/license-Apache%202.0-blue.svg)](https://github.com/JetBrains/databao-context-engine?tab=License-1-ov-file)
+
+[//]: # ([![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 is a CLI tool that **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             |
+
+## Installation
+
+Databao Context Engine is [available on PyPI](https://pypi.org/project/databao-context-engine/) and can be installed with uv, pip, or another package manage.
+
+### Using uv
+
+1. Install Databao Context Engine:
+
+   ```bash
+   uv tool install databao-context-engine
+   ```
+
+1. Add it to your PATH:
+
+   ```bash
+   uv tool update-shell
+   ```
+
+1. Verify the installation:
+
+   ```bash
+   dce --help
+   ```
+
+### Using pip
+
+1. Install Databao Context Engine:
+
+   ```bash
+   pip install databao-context-engine
+   ```
+
+1. Verify the installation:
+
+   ```bash
+   dce --help
+   ```
+
+##  Supported data sources
+
+* <img src="https://cdn.simpleicons.org/postgresql/316192" width="16" height="16" alt=""> PostgreSQL
+* <img src="https://cdn.simpleicons.org/mysql/4479A1" width="16" height="16" alt=""> MySQL
+* <img src="https://cdn.simpleicons.org/sqlite/003B57" width="16" height="16" alt=""> SQLite
+* <img src="https://cdn.simpleicons.org/duckdb/FFF000" width="16" height="16" alt=""> DuckDB
+* <img src="https://cdn.simpleicons.org/dbt/FF694B" width="16" height="16" alt=""> dbt projects
+* 📄 Documents & spreadsheets *(coming soon)*
+
+##  Supported LLMs
+
+| Provider      | Configuration                                |
+|---------------|----------------------------------------------|
+| **Ollama**    | `languageModel: OLLAMA`: runs locally, free  |
+| **OpenAI**    | `languageModel: OPENAI`: requires an API key |
+| **Anthropic** | `languageModel: CLAUDE`: requires an API key |
+| **Google**    | `languageModel: GEMINI`: requires an API key |
+
+## Quickstart
+
+### 1. Create a project
+
+1. Create a new directory for your project and navigate to it:
+
+   ```bash
+   mkdir dce-project && cd dce-project
+   ```
+
+1. Initialize a new project:
+
+   ```bash
+   dce init
+   ```
+
+### 2. Configure data sources
+
+1. When prompted, agree to create a new datasource.
+   You can also use the `dce datasource add` command.
+
+1. Provide the data source type and its name.
+
+1. Open the config file that was created for you in your editor and fill in the connection details.
+
+1. Repeat these steps for all data sources you want to include in your project.
+
+1. If you have data in Markdown or text files,
+   you can add them to the `dce/src/files` directory.
+
+### 3. Build context
+
+1. To build the context, run the following command:
+
+   ```bash
+   dce build
+   ```
+
+### 4. Use Context with Your LLM
+
+**Option A: Dynamic via MCP Server**
+
+Databao Context Engine exposes the context through a local MCP Server, so your agent can access the latest context at runtime.
+
+1. In **Claude Desktop**, **Cursor**, or another MCP-compatible agent, add the following configuration.
+   Replace `dce-project/` with the path to your project directory:
+  
+   ```json 
+   # claude_desktop_config.json, mcp.json, or similar
+   
+   {
+     "mcpServers": {
+       "dce": {
+         "command": "dce mcp",
+         "args": ["--project-dir", "dce-project/"]
+       }
+     }
+   }
+   ```
+
+1. Save the file and restart your agent.
+
+1. Open a new chat, in the chat window, select the `dce` server, and ask questions related to your project context.
+
+**Option B: Static artifact**
+
+Even if you don’t have Claude or Cursor installed on your local machine,
+you can still use the context built by Databao Context Engine by pasting it directly into your chat with an AI assistant.
+
+1. Navigate to `dce-project/output/` and open the directory with the latest run.
+
+1. Attach the `all_results.yaml` file to your chat with the AI assistant or copy and paste its contents into your chat.
+
+##  Contributing
+
+We’d love your help! Here’s how to get involved:
+
+- ⭐ **Star this repo** — it helps others find us!
+- 🐛 **Found a bug?** [Open an issue](https://github.com/JetBrains/databao-context-engine/issues)
+- 💡 **Have an idea?** We’re all ears — create a feature request
+- 👍 **Upvote issues** you care about — helps us prioritize
+- 🔧 **Submit a PR**
+- 📝 **Improve docs** — typos, examples, tutorials — everything helps!
+
+New to open source? No worries! We're friendly and happy to help you get started. 🌱
+
+For more details, see [CONTRIBUTING](CONTRIBUTING.md).
+
+## 📄 License
+
+Apache 2.0 — use it however you want. See the [LICENSE](LICENSE.md) file for details.
+
+---
+
+<p align="center">
+ <b>Like Databao Context Engine?</b> Give us a ⭐ — it means a lot!
+</p>
+
+<p align="center">
+ <a href="https://databao.app">Website</a> •
+ <a href="https://discord.gg/databao">Discord</a>
+</p>

{databao_context_engine-0.1.4.dev1 → databao_context_engine-0.1.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "databao-context-engine"
-version = "0.1.4.dev1"
+version = "0.1.6"
 description = "Semantic context for your LLMs — generated automatically"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -48,7 +48,7 @@ dev = [
     "pre-commit>=4.5.1",
     "pytest>=9.0.2",
     "pytest-unordered>=0.7.0",
-    "ruff>=0.14.2",
+    "ruff>=0.14.4",
     "testcontainers==4.12.0",
     "types-pyyaml>=6.0.12.20250915",
     "types-pymysql>=1.1.0.20250916",
@@ -66,6 +66,18 @@ override-dependencies = ["urllib3>=2.6.3", "filelock>=3.20.3", "virtualenv>=20.3
 [tool.uv.build-backend]
 source-exclude = ["tests"]
 
+[tool.mypy]
+#strict = true
+show_error_codes = true
+check_untyped_defs = false
+disallow_untyped_defs = false ## many errors now
+warn_redundant_casts = true
+warn_unused_configs = true
+warn_unused_ignores = false
+allow_redefinition_new = true
+local_partial_types = true
+files = ["src", "tests"]
+
 [[tool.mypy.overrides]]
 module = ["requests", "requests.*"]
 ignore_missing_imports = true

{databao_context_engine-0.1.4.dev1 → databao_context_engine-0.1.6}/src/databao_context_engine/__init__.py

@@ -8,6 +8,13 @@ from databao_context_engine.datasources.check_config import (
 from databao_context_engine.datasources.datasource_context import DatasourceContext
 from databao_context_engine.datasources.types import Datasource, DatasourceId
 from databao_context_engine.init_project import init_dce_project, init_or_get_dce_project
+from databao_context_engine.llm import (
+    OllamaError,
+    OllamaPermanentError,
+    OllamaTransientError,
+    download_ollama_models_if_needed,
+    install_ollama_if_needed,
+)
 from databao_context_engine.plugin_loader import DatabaoContextPluginLoader
 from databao_context_engine.pluginlib.build_plugin import (
     BuildDatasourcePlugin,
@@ -16,7 +23,7 @@ from databao_context_engine.pluginlib.build_plugin import (
     DatasourceType,
 )
 from databao_context_engine.pluginlib.config import ConfigPropertyDefinition
-from databao_context_engine.project.info import DceInfo, get_databao_context_engine_info
+from databao_context_engine.project.info import DceInfo, DceProjectInfo, get_databao_context_engine_info
 from databao_context_engine.project.init_project import InitErrorReason, InitProjectError
 from databao_context_engine.services.chunk_embedding_service import ChunkEmbeddingMode
 
@@ -35,6 +42,7 @@ __all__ = [
     "DatasourceType",
     "get_databao_context_engine_info",
     "DceInfo",
+    "DceProjectInfo",
     "init_dce_project",
     "init_or_get_dce_project",
     "InitErrorReason",
@@ -44,4 +52,9 @@ __all__ = [
     "BuildPlugin",
     "BuildDatasourcePlugin",
     "BuildFilePlugin",
+    "install_ollama_if_needed",
+    "download_ollama_models_if_needed",
+    "OllamaError",
+    "OllamaTransientError",
+    "OllamaPermanentError",
 ]

{databao_context_engine-0.1.4.dev1 → databao_context_engine-0.1.6}/src/databao_context_engine/build_sources/build_runner.py

@@ -13,7 +13,7 @@ from databao_context_engine.datasources.datasource_discovery import discover_dat
 from databao_context_engine.datasources.types import DatasourceId
 from databao_context_engine.pluginlib.build_plugin import DatasourceType
 from databao_context_engine.plugins.plugin_loader import load_plugins
-from databao_context_engine.project.layout import get_output_dir
+from databao_context_engine.project.layout import ProjectLayout
 
 logger = logging.getLogger(__name__)
 
@@ -36,7 +36,7 @@ class BuildContextResult:
 
 
 def build(
-    project_dir: Path,
+    project_layout: ProjectLayout,
     *,
     build_service: BuildService,
 ) -> list[BuildContextResult]:
@@ -55,15 +55,15 @@ def build(
     """
     plugins = load_plugins()
 
-    datasources = discover_datasources(project_dir)
+    datasources = discover_datasources(project_layout)
 
     if not datasources:
-        logger.info("No sources discovered under %s", project_dir)
+        logger.info("No sources discovered under %s", project_layout.src_dir)
         return []
 
     number_of_failed_builds = 0
     build_result = []
-    reset_all_results(get_output_dir(project_dir))
+    reset_all_results(project_layout.output_dir)
     for discovered_datasource in datasources:
         try:
             prepared_source = prepare_source(discovered_datasource)
@@ -87,14 +87,14 @@ def build(
                 plugin=plugin,
             )
 
-            output_dir = get_output_dir(project_dir)
+            output_dir = project_layout.output_dir
 
             context_file_path = export_build_result(output_dir, result)
             append_result_to_all_results(output_dir, result)
 
             build_result.append(
                 BuildContextResult(
-                    datasource_id=DatasourceId.from_string_repr(result.datasource_id),
+                    datasource_id=discovered_datasource.datasource_id,
                     datasource_type=DatasourceType(full_type=result.datasource_type),
                     context_built_at=result.context_built_at,
                     context_file_path=context_file_path,

{databao_context_engine-0.1.4.dev1 → databao_context_engine-0.1.6}/src/databao_context_engine/build_sources/build_wiring.py

@@ -1,5 +1,4 @@
 import logging
-from pathlib import Path
 
 from duckdb import DuckDBPyConnection
 
@@ -12,7 +11,7 @@ from databao_context_engine.llm.factory import (
     create_ollama_embedding_provider,
     create_ollama_service,
 )
-from databao_context_engine.project.layout import ensure_project_dir
+from databao_context_engine.project.layout import ProjectLayout
 from databao_context_engine.services.chunk_embedding_service import ChunkEmbeddingMode
 from databao_context_engine.services.factories import create_chunk_embedding_service
 from databao_context_engine.storage.connection import open_duckdb_connection
@@ -22,7 +21,9 @@ from databao_context_engine.system.properties import get_db_path
 logger = logging.getLogger(__name__)
 
 
-def build_all_datasources(project_dir: Path, chunk_embedding_mode: ChunkEmbeddingMode) -> list[BuildContextResult]:
+def build_all_datasources(
+    project_layout: ProjectLayout, chunk_embedding_mode: ChunkEmbeddingMode
+) -> list[BuildContextResult]:
     """Build the context for all datasources in the project.
 
     - Instantiates the build service
@@ -31,21 +32,18 @@ def build_all_datasources(project_dir: Path, chunk_embedding_mode: ChunkEmbeddin
     Returns:
         A list of all the contexts built.
     """
-    ensure_project_dir(project_dir)
-
-    logger.debug(f"Starting to build datasources in project {project_dir.resolve()}")
+    logger.debug(f"Starting to build datasources in project {project_layout.project_dir.resolve()}")
 
     # Think about alternative solutions. This solution will mirror the current behaviour
     # The current behaviour only builds what is currently in the /src folder
     # This will need to change in the future when we can pick which datasources to build
-    db_path = get_db_path(project_dir)
+    db_path = get_db_path(project_layout.project_dir)
     db_path.parent.mkdir(parents=True, exist_ok=True)
     if db_path.exists():
         db_path.unlink()
 
+    migrate(db_path)
     with open_duckdb_connection(db_path) as conn:
-        migrate(db_path)
-
         ollama_service = create_ollama_service()
         embedding_provider = create_ollama_embedding_provider(ollama_service)
         description_provider = (
@@ -60,7 +58,7 @@ def build_all_datasources(project_dir: Path, chunk_embedding_mode: ChunkEmbeddin
             chunk_embedding_mode=chunk_embedding_mode,
         )
         return build(
-            project_dir=project_dir,
+            project_layout=project_layout,
             build_service=build_service,
         )
 

{databao_context_engine-0.1.4.dev1 → databao_context_engine-0.1.6}/src/databao_context_engine/build_sources/plugin_execution.py

@@ -2,7 +2,7 @@ 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.datasources.types import PreparedConfig, PreparedDatasource
 from databao_context_engine.pluginlib.build_plugin import (
     BuildDatasourcePlugin,
     BuildFilePlugin,
@@ -39,11 +39,8 @@ class BuiltDatasourceContext:
 
 def execute(prepared_datasource: PreparedDatasource, plugin: BuildPlugin) -> BuiltDatasourceContext:
     built_context = _execute(prepared_datasource, plugin)
-
-    datasource_id = DatasourceId.from_datasource_config_file_path(prepared_datasource.path)
-
     return BuiltDatasourceContext(
-        datasource_id=str(datasource_id),
+        datasource_id=str(prepared_datasource.datasource_id),
         datasource_type=prepared_datasource.datasource_type.full_type,
         context_built_at=datetime.now(),
         context=built_context,
@@ -61,10 +58,10 @@ def _execute(prepared_datasource: PreparedDatasource, plugin: BuildPlugin) -> An
             config=prepared_datasource.config,
             datasource_name=prepared_datasource.datasource_name,
         )
-    else:
-        file_plugin = cast(BuildFilePlugin, plugin)
-        return execute_file_plugin(
-            plugin=file_plugin,
-            datasource_type=prepared_datasource.datasource_type,
-            file_path=prepared_datasource.path,
-        )
+
+    file_plugin = cast(BuildFilePlugin, plugin)
+    return execute_file_plugin(
+        plugin=file_plugin,
+        datasource_type=prepared_datasource.datasource_type,
+        file_path=prepared_datasource.path,
+    )