wip

test_lite_agent.py

@@ -0,0 +1,65 @@
+from crewai import LLM
+from crewai.lite_agent import LiteAgent
+from crewai.tools import BaseTool
+
+
+# A simple test tool
+class TestTool(BaseTool):
+    name = "test_tool"
+    description = "A simple test tool"
+
+    def _run(self, query: str) -> str:
+        return f"Test result for: {query}"
+
+
+# Test with tools
+def test_with_tools():
+    llm = LLM(model="gpt-4o")
+    agent = LiteAgent(
+        role="Test Agent",
+        goal="Test the system prompt formatting",
+        backstory="I am a test agent created to verify the system prompt works correctly.",
+        llm=llm,
+        tools=[TestTool()],
+        verbose=True,
+    )
+
+    # Get the system prompt
+    system_prompt = agent._get_default_system_prompt()
+    print("\n=== System Prompt (with tools) ===")
+    print(system_prompt)
+
+    # Test a simple query
+    response = agent.kickoff("Hello, can you help me?")
+    print("\n=== Agent Response ===")
+    print(response)
+
+
+# Test without tools
+def test_without_tools():
+    llm = LLM(model="gpt-4o")
+    agent = LiteAgent(
+        role="Test Agent",
+        goal="Test the system prompt formatting",
+        backstory="I am a test agent created to verify the system prompt works correctly.",
+        llm=llm,
+        verbose=True,
+    )
+
+    # Get the system prompt
+    system_prompt = agent._get_default_system_prompt()
+    print("\n=== System Prompt (without tools) ===")
+    print(system_prompt)
+
+    # Test a simple query
+    response = agent.kickoff("Hello, can you help me?")
+    print("\n=== Agent Response ===")
+    print(response)
+
+
+if __name__ == "__main__":
+    print("Testing LiteAgent with tools...")
+    test_with_tools()
+
+    print("\n\nTesting LiteAgent without tools...")
+    test_without_tools()