Feature/poc

.env.example

@@ -0,0 +1,4 @@
+OPENAI_API_KEY=''
+MODEL=''
+MODEL_PROVIDER=''
+

.github/workflows/format.yml

@@ -7,12 +7,12 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        linter: [ruff, mypy]
+        linter: [ruff] #, mypy]
         include:
           - linter: ruff
-            command: ruff check --fix .
-          - linter: mypy
-            command: mypy .
+            command: ruff check --fix . --exclude ./notebook/
+          # - linter: mypy
+          #   command: mypy . --exclude ./notebook/
     steps:
       - uses: actions/checkout@v4
       - name: Install uv

.gitignore

@@ -1,15 +1,21 @@
 # Created by https://www.toptal.com/developers/gitignore/api/python
 # Edit at https://www.toptal.com/developers/gitignore?templates=python
-
+*.md
 ### Python ###
 # Byte-compiled / optimized / DLL files
 __pycache__/
 *.py[cod]
+myenv/
 *$py.class
 .idea
 # C extensions
 *.so
-
+.env
+/notebook/logs
+*.csv
+*.ipynb
+*.db
+*.Identifier
 # Distribution / packaging
 .Python
 build/

.pre-commit-config.yaml

@@ -2,7 +2,7 @@ repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
     rev: v4.6.0
     hooks:
-      - id: check-added-large-files
+      # - id: check-added-large-files
       - id: check-ast
       - id: check-builtin-literals
       - id: check-case-conflict
@@ -19,6 +19,7 @@ repos:
       - id: detect-private-key
       - id: end-of-file-fixer
         types: [python]
+        exclude: ^notebook/
       - id: fix-byte-order-marker
       - id: mixed-line-ending
       - id: name-tests-test
@@ -30,7 +31,7 @@ repos:
     rev: v3.16.0
     hooks:
     - id: pyupgrade
-      args: [--py312-plus]
+      args: [--py39-plus]
 
   - repo: local
     hooks:
@@ -39,19 +40,29 @@ repos:
         entry: uv run ruff check --fix
         language: system
         types: [python]
+        exclude: ^notebook/
 
       - id: ruff-format
         name: Python code formatting with Ruff
         entry: uv run ruff format
         language: system
         types: [python]
+        exclude: ^notebook/
 
-      - id: mypy
-        name: Python type checking with MyPy
-        entry: uv run mypy
-        language: system
-        types: [python]
-        pass_filenames: false
-        args:
-        - "app"
-        - "tests"
+  #     - id: mypy
+  #       name: Python type checking with MyPy
+  #       entry: uv run mypy
+  #       language: system
+  #       types: [python]
+  #       pass_filenames: false
+  #       args:
+  #       - "app"
+  #       - "tests"
+  #       exclude: ^notebook/
+  # -   repo: local
+  #     hooks:
+  #     -   id: pyright
+  #         name: pyright
+  #         entry: pyright
+  #         language: system
+  #         types: [python]

.python-version

@@ -0,0 +1 @@
+3.12

Dockerfile

@@ -14,4 +14,10 @@ COPY ./app /code/app
 RUN groupadd -r nonroot && useradd -r -g nonroot nonroot
 USER nonroot
 
-CMD ["/code/.venv/bin/uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000"]
+CMD ["/code/.venv/bin/uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+# To build the Docker image, use the following command:
+# docker build -t my-fastapi-app .
+
+# To run the Docker container, use the following command:
+# docker run -p 8000:8000 my-fastapi-app
+

Makefile

@@ -0,0 +1,28 @@
+pytest:
+	pytest --cov-report term --cov=app ./tests
+
+pre-commit:
+	pre-commit run --all-files
+
+clean:
+	find . | grep -E "(__pycache__|\.pyc|\.pyo)" | xargs rm -rf
+	find . | grep -E ".pytest_cache" | xargs rm -rf
+	find . | grep -E ".ipynb_checkpoints" | xargs rm -rf
+	rm -rf .coverage
+
+ruff-check:
+	ruff check --fix .
+
+ruff-format:
+	ruff format .
+
+ruff: ruff-check ruff-format
+
+run:
+	uv run uvicorn main:app --port 8000 --reload
+
+ops: pytest pre-commit clean ruff
+	@echo "\033[92mAll operations completed successfully.\033[0m"
+	@echo "\033[93mPlease check the output above for any errors or warnings.\033[0m"
+	@echo "\033[93mIf you see any issues, please address them before proceeding.\033[0m"
+	@echo "\033[92mThank you for using this Makefile!\033[0m"

README.md

@@ -23,60 +23,7 @@
   </a>
 </div>
 
-# Python FastAPI Template
-
-A template for a Python FastAPI service:
-- [x] [uv](https://docs.astral.sh/uv/getting-started/) for dependency management
-- [x] [FastAPI](https://fastapi.tiangolo.com/) for building the API
-- [x] [Devcontainer](https://code.visualstudio.com/docs/devcontainers/tutorial) for development environment
-- [x] [Docker](https://www.docker.com/) for containerization
-- [x] [Black](https://black.readthedocs.io/) & [isort](https://pycqa.github.io/isort/) for code formatting
-- [x] [pre-commit](https://pre-commit.com/) for code quality checks
-- [x] [Github Actions](https://github.com/features/actions) for CI/CD
-- [x] [pytest](https://docs.pytest.org/) for testing
-
-## Build and run locally
-
-Install `uv` and `python3.11`
-```
-curl -LsSf https://astral.sh/uv/install.sh | sh
-uv python install 3.11
-```
-
-Then install dependencies
-```
-uv sync --frozen
-```
-
-Run locally
-```
-uv run uvicorn app.main:app --port 8000 --reload
-```
-
-## Run tests
-```
-uv run pytest
-```
-
-## Build and run using Docker
-
-```
-docker build -t app .
-docker run -p 8000:8000 -it app
-```
-
-### Notes
-- If you face an issue with **git ssh access** while pushing new changes, run `ssh-add $HOME/.ssh/<your ssh key>` in terminal outside the devcontainer.
-
-- If you face an issue during **devcontainer build**, make sure the repo is marked as trusted in VSCode. Check `Source Control` tab in the sidebar to mark the repo safe, then rebuild the devcontainer.
-
-## Sample CRUD API
-
-The `/v1` directory contains a sample API router demonstrating basic CRUD operations for users:
-
 - Endpoints: Create, Read, Update, Delete users
 - Router setup: `app/v1/routers/base.py` and `app/v1/routers/users.py`
 - User model: `app/v1/models/user.py`
 - User management: `app/v1/services/user/user_manager.py`
-
-Use the samples as a starting point for your own API endpoints. View available endpoints at `http://localhost:8000/docs`.

app/__init__.py

@@ -0,0 +1,16 @@
+# app/__init__.py
+
+from .agent_orchestrator import determine_data_source, process_user_input_stream, run_agent_orchestrator
+from .tools.data_analyst_agent import DataVisualizationAgent
+from .tools.sql_data_analyst_agent import SQLDataAnalysisAgent
+from .visualization_server import VisualizationServer
+
+
+__all__ = [
+    "determine_data_source",
+    "process_user_input_stream",
+    "run_agent_orchestrator",
+    "VisualizationServer",
+    "DataVisualizationAgent",
+    "SQLDataAnalysisAgent",
+]