47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script for OpenClaw Agent
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
sys.path.insert(0, '.')
|
|
|
|
from main import create_maxclaw_agent
|
|
|
|
def test_agent_creation():
|
|
"""Test that the agent can be created successfully."""
|
|
print("Testing agent creation...")
|
|
try:
|
|
agent = create_maxclaw_agent()
|
|
print("✅ Agent created successfully!")
|
|
return agent
|
|
except Exception as e:
|
|
print(f"❌ Failed to create agent: {e}")
|
|
print("💡 Make sure to set the required environment variables for your chosen model provider.")
|
|
print(" See README.md for configuration examples.")
|
|
return None
|
|
|
|
def test_simple_query(agent):
|
|
"""Test a simple query to the agent."""
|
|
print("\nTesting simple query...")
|
|
try:
|
|
result = agent.invoke({"messages": [{"role": "user", "content": "Hello, how are you?"}]})
|
|
ai_message = result["messages"][-1]
|
|
print(f"✅ Agent responded: {ai_message.content}")
|
|
return True
|
|
except Exception as e:
|
|
print(f"❌ Failed to get response: {e}")
|
|
return False
|
|
|
|
if __name__ == "__main__":
|
|
# Set some default environment variables for testing
|
|
# In a real scenario, these would be set externally
|
|
if not os.getenv("MODEL_PROVIDER"):
|
|
os.environ["MODEL_PROVIDER"] = "openai"
|
|
if not os.getenv("MODEL_NAME"):
|
|
os.environ["MODEL_NAME"] = "gpt-4o-mini" # Using a cheaper model for testing
|
|
|
|
agent = test_agent_creation()
|
|
if agent:
|
|
test_simple_query(agent) |