add tests

run-llama · Jun 4, 2024 · 83c56f4 · 83c56f4
1 parent 64b0064
commit 83c56f4
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 0 deletions.
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/__pycache__/__init__.cpython-311.pyc b/tests/__pycache__/__init__.cpython-311.pyc
diff --git a/tests/__pycache__/test_agent_server.cpython-311-pytest-8.2.2.pyc b/tests/__pycache__/test_agent_server.cpython-311-pytest-8.2.2.pyc
diff --git a/tests/test_agent_server.py b/tests/test_agent_server.py
@@ -0,0 +1,16 @@
+from llama_index.core.llms import MockLLM
+from llama_index.core.agent import ReActAgent
+
+from agentfile.agent import AgentServer
+
+
+def test_init() -> None:
+    agent = ReActAgent.from_tools([], llm=MockLLM())
+    server = AgentServer(
+        agent, running=False, description="Test Agent Server", step_interval=0.5
+    )
+
+    assert server.agent == agent
+    assert server.running is False
+    assert server.description == "Test Agent Server"
+    assert server.step_interval == 0.5