diff --git a/pyproject.toml.jinja b/pyproject.toml.jinja index 762dd75..08d2618 100644 --- a/pyproject.toml.jinja +++ b/pyproject.toml.jinja @@ -19,7 +19,22 @@ requires = ["hatchling"] build-backend = "hatchling.build" [dependency-groups] -dev = [] +dev = [ + "hatch>=1.14.1", + "pytest>=8.4.2", + "ruff>=0.13.0", + "ty>=0.0.1a20", +] + +[tool.hatch.envs.default.scripts] +"format" = "ruff format ." +"format-check" = "ruff format --check ." +"lint" = "ruff check --fix ." +"lint-check" = ["ruff check ."] +typecheck = "ty check src" +test = "pytest" +"all-check" = ["format-check", "lint-check", "test"] +"all-fix" = ["format", "lint", "test"] [tool.llamadeploy] env-files = [".env"] diff --git a/src/{{ project_name_snake }}/qa_workflows.py b/src/{{ project_name_snake }}/qa_workflows.py index cde304a..5363255 100644 --- a/src/{{ project_name_snake }}/qa_workflows.py +++ b/src/{{ project_name_snake }}/qa_workflows.py @@ -1,6 +1,5 @@ import logging import os -import uuid import httpx from llama_cloud.types import RetrievalMode @@ -23,7 +22,6 @@ from llama_index.embeddings.openai import OpenAIEmbedding from llama_index.core.memory import ChatMemoryBuffer from .clients import ( - INDEX_NAME, LLAMA_CLOUD_API_KEY, LLAMA_CLOUD_BASE_URL, get_custom_client, @@ -335,5 +333,6 @@ class ChatWorkflow(Workflow): result={"success": False, "error": f"Error processing query: {str(e)}"} ) + upload = DocumentUploadWorkflow(timeout=None) -chat = ChatWorkflow(timeout=None) \ No newline at end of file +chat = ChatWorkflow(timeout=None) diff --git a/test-proj/pyproject.toml b/test-proj/pyproject.toml index 59ce3a5..7206b7e 100644 --- a/test-proj/pyproject.toml +++ b/test-proj/pyproject.toml @@ -19,7 +19,22 @@ requires = ["hatchling"] build-backend = "hatchling.build" [dependency-groups] -dev = [] +dev = [ + "hatch>=1.14.1", + "pytest>=8.4.2", + "ruff>=0.13.0", + "ty>=0.0.1a20", +] + +[tool.hatch.envs.default.scripts] +"format" = "ruff format ." +"format-check" = "ruff format --check ." +"lint" = "ruff check --fix ." +"lint-check" = ["ruff check ."] +typecheck = "ty check src" +test = "pytest" +"all-check" = ["format-check", "lint-check", "test"] +"all-fix" = ["format", "lint", "test"] [tool.llamadeploy] env-files = [".env"] diff --git a/test-proj/src/test_proj/config.py b/test-proj/src/test_proj/config.py index 0df6766..28c04d7 100644 --- a/test-proj/src/test_proj/config.py +++ b/test-proj/src/test_proj/config.py @@ -1,2 +1,2 @@ PROJECT_ID = "asdf" -ORGANIZATION_ID = "asdf" \ No newline at end of file +ORGANIZATION_ID = "asdf" diff --git a/test-proj/src/test_proj/qa_workflows.py b/test-proj/src/test_proj/qa_workflows.py index cde304a..5363255 100644 --- a/test-proj/src/test_proj/qa_workflows.py +++ b/test-proj/src/test_proj/qa_workflows.py @@ -1,6 +1,5 @@ import logging import os -import uuid import httpx from llama_cloud.types import RetrievalMode @@ -23,7 +22,6 @@ from llama_index.embeddings.openai import OpenAIEmbedding from llama_index.core.memory import ChatMemoryBuffer from .clients import ( - INDEX_NAME, LLAMA_CLOUD_API_KEY, LLAMA_CLOUD_BASE_URL, get_custom_client, @@ -335,5 +333,6 @@ class ChatWorkflow(Workflow): result={"success": False, "error": f"Error processing query: {str(e)}"} ) + upload = DocumentUploadWorkflow(timeout=None) -chat = ChatWorkflow(timeout=None) \ No newline at end of file +chat = ChatWorkflow(timeout=None)