80 lines
No EOL
2.8 KiB
Python
80 lines
No EOL
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Test script to verify TALESTORM_AGENT_ID usage
|
|
"""
|
|
|
|
import asyncio
|
|
from src.services.session_service import session_service
|
|
from src.config import settings
|
|
|
|
async def test_agent_id_usage():
|
|
"""Test that the configured agent ID is being used"""
|
|
print("Testing TALESTORM_AGENT_ID Usage...")
|
|
print("=" * 50)
|
|
|
|
print(f"Configured Agent ID: {settings.TALESTORM_AGENT_ID}")
|
|
print(f"Session Service Agent ID: {session_service.agent_id}")
|
|
|
|
# Test session creation
|
|
print("\nCreating session with configured agent...")
|
|
session_id = await session_service.create_session(agent_id=settings.TALESTORM_AGENT_ID)
|
|
|
|
if session_id:
|
|
print(f"✅ Session created: {session_id}")
|
|
|
|
# Get session details to verify agent
|
|
session_details = await session_service.get_session(session_id)
|
|
if session_details:
|
|
print(f"Session Agent ID: {session_details.get('agent_id')}")
|
|
print(f"Session Organization ID: {session_details.get('organization_id')}")
|
|
else:
|
|
print("⚠️ Could not retrieve session details")
|
|
else:
|
|
print("❌ Failed to create session")
|
|
|
|
# Test listing agents
|
|
print("\nListing available agents...")
|
|
agents = await session_service.list_agents()
|
|
if agents:
|
|
print(f"✅ Found {len(agents)} agents:")
|
|
for i, agent in enumerate(agents, 1):
|
|
print(f" {i}. {agent.get('name', 'Unknown')} (ID: {agent.get('id')})")
|
|
|
|
# Highlight the configured agent
|
|
if str(agent.get('id')) == settings.TALESTORM_AGENT_ID:
|
|
print(f" ⭐ This is the configured agent!")
|
|
else:
|
|
print("❌ No agents found")
|
|
|
|
print("\n✅ Agent ID test completed!")
|
|
|
|
async def test_agent_validation():
|
|
"""Test that the configured agent exists and is accessible"""
|
|
print("\nTesting Agent Validation...")
|
|
print("=" * 50)
|
|
|
|
if not settings.TALESTORM_AGENT_ID:
|
|
print("❌ TALESTORM_AGENT_ID not configured")
|
|
return
|
|
|
|
# Try to get the configured agent
|
|
agent = await session_service.get_default_agent()
|
|
if agent:
|
|
print(f"✅ Configured agent found:")
|
|
print(f" Name: {agent.get('name', 'Unknown')}")
|
|
print(f" ID: {agent.get('id')}")
|
|
print(f" Model: {agent.get('model_name', 'Unknown')}")
|
|
print(f" Provider: {agent.get('model_provider', 'Unknown')}")
|
|
print(f" RAG Enabled: {agent.get('enable_rag_search', False)}")
|
|
else:
|
|
print("❌ Configured agent not found or not accessible")
|
|
|
|
print("\n✅ Agent validation completed!")
|
|
|
|
async def main():
|
|
"""Run all tests"""
|
|
await test_agent_id_usage()
|
|
await test_agent_validation()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |