Merge pull request #15 from agntcy/PUCCINI-270

Add unit tests covering basic features

Author: mtrinell

Diff for: .env.example

@@ -1,11 +1,9 @@
-# POETRY ENV
-POETRY_VIRTUALENVS_IN_PROJECT=true
-
 ### SERVER-SPECIFIC ENV ###
 API_HOST=127.0.0.1
 API_PORT=8000
 AGENTS_REF='{"agent_name": "agent_module_name:agent_var"}'
 AGENT_MANIFEST_PATH=manifest.json
+AGWS_STORAGE_PERSIST=True
 AGWS_STORAGE_PATH=agws_storage.pkl
 NUM_WORKERS=5
 API_KEY=your-secret-key-here

Diff for: .vscode/settings.json

@@ -0,0 +1,14 @@
+{
+    "python.testing.pytestArgs": [
+        "tests"
+    ],
+    "python.testing.unittestEnabled": false,
+    "python.testing.pytestEnabled": true,
+    "[python]": {
+        "editor.formatOnSave": true,
+        "editor.defaultFormatter": "charliermarsh.ruff",
+        "editor.codeActionsOnSave": {
+            "source.fixAll": "explicit"
+        }
+    }
+}

Diff for: Makefile

@@ -7,7 +7,7 @@ PACKAGE_NAME := agent_workflow_server.generated
 GENERATOR_IMAGE := openapitools/openapi-generator-cli:latest
 ADDITIONAL_PROPERTIES := packageName=$(PACKAGE_NAME),python_typed=true
 
-.PHONY: clean validate-spec update-spec generate-api run docker-build-dev
+.PHONY: clean validate-spec update-spec generate-api run docker-build-dev test format lint lint-fix
 
 # Ensure output directory exists
 $(OUTPUT_DIR):
@@ -46,5 +46,17 @@ run:
 	poetry install
 	poetry run server
 
+test:
+	poetry run pytest
+
+format:
+	poetry run ruff format .
+
+lint:
+	poetry run ruff check .
+
+lint-fix:
+	poetry run ruff check . --fix
+
 docker-build-dev: ## Build the docker image.
 	docker buildx bake workflowserver-dev --load

Diff for: poetry.lock

@@ -0,0 +1,2 @@
+[virtualenvs]
+in-project = true

Diff for: poetry.toml

@@ -15,12 +15,35 @@ dependencies = [
     "uvicorn (>=0.34.0,<0.35.0)",
     "langgraph (>=0.2.60,<0.4.0)",
     "llama-index (>=0.12.0,<0.13.0)",
-    "dotenv (>=0.9.9,<0.10.0)"
+    "dotenv (>=0.9.9,<0.10.0)",
 ]
 
 [build-system]
 requires = ["poetry-core>=2.0.0,<3.0.0"]
 build-backend = "poetry.core.masonry.api"
 
 [tool.poetry.scripts]
-server = "agent_workflow_server.main:start"
+server = "agent_workflow_server.main:start"
+
+[tool.poetry.group.dev.dependencies]
+pytest = "^8.3.5"
+pytest-mock = "^3.14.0"
+ruff = "^0.11.0"
+pytest-asyncio = "^0.25.3"
+
+[tool.ruff]
+lint.select = [ "E", "F", "I", "TID251"]
+lint.ignore = [ "E501" ]
+indent-width = 4
+exclude = ["src/agent_workflow_server/generated/*"]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"
+
+[tool.pytest.ini_options]
+minversion = "6.0"
+testpaths = [
+    "tests"
+]
+asyncio_default_fixture_loop_scope="function"

Diff for: pyproject.toml

@@ -1,7 +1,8 @@
 from typing import Optional
+
 from langgraph.graph.graph import CompiledGraph, Graph
 
-from agent_workflow_server.agents.base import BaseAgent, BaseAdapter
+from agent_workflow_server.agents.base import BaseAdapter, BaseAgent
 
 
 class LangGraphAdapter(BaseAdapter):
@@ -18,5 +19,7 @@ def __init__(self, agent: CompiledGraph):
         self.agent = agent
 
     async def astream(self, input: dict, config: dict):
-        async for event in self.agent.astream(input=input, config=config, stream_mode="values"):
+        async for event in self.agent.astream(
+            input=input, config=config, stream_mode="values"
+        ):
             yield event

Diff for: src/agent_workflow_server/agents/adapters/langgraph.py

@@ -1,7 +1,10 @@
+import inspect
 from typing import Optional
+
 from llama_index.core.workflow import Workflow
-from agent_workflow_server.agents.base import BaseAgent, BaseAdapter
-import inspect
+
+from agent_workflow_server.agents.base import BaseAdapter, BaseAgent
+
 
 class LlamaIndexAdapter(BaseAdapter):
     def load_agent(self, agent: object) -> Optional[BaseAgent]:

Diff for: src/agent_workflow_server/agents/adapters/llamaindex.py

@@ -1,5 +1,5 @@
 from abc import ABC, abstractmethod
-from typing import AsyncGenerator, Optional, Any
+from typing import Any, AsyncGenerator, Optional
 
 
 class BaseAgent(ABC):