File size: 4,484 Bytes
ff0e97f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 |
"""
Test structured output parsing for agent responses.
Tests the parse_agent_response function to ensure it correctly formats
images, audio, and species information from raw agent text.
"""
import asyncio
import sys
from pathlib import Path
# Add parent directory to sys.path
parent_dir = Path(__file__).parent.parent
sys.path.insert(0, str(parent_dir))
from langgraph_agent.structured_output import parse_agent_response
async def test_structured_output():
"""Test structured output parsing with various response formats."""
print("=" * 60)
print("TESTING STRUCTURED OUTPUT PARSING")
print("=" * 60)
# Test Case 1: Response with images and species identification
print("\n[TEST 1]: Response with images and species")
print("-" * 40)
test_response_1 = """
Based on the image, I can identify this as a Northern Cardinal. The bright red plumage and distinctive crest are characteristic of this species.
Here are some reference images:
The Northern Cardinal is a beautiful songbird commonly found in North America.
"""
result_1 = await parse_agent_response(
raw_response=test_response_1,
provider="openai",
api_key="test-key",
model="gpt-4o-mini"
)
print("Input length:", len(test_response_1))
print("Output length:", len(result_1))
print("Contains markdown images:", "![Northern Cardinal]" in result_1)
print("Contains species name:", "Northern Cardinal" in result_1)
print("✅ Test 1 completed")
# Test Case 2: Response with audio recordings
print("\n[TEST 2]: Response with audio recordings")
print("-" * 40)
test_response_2 = """
This appears to be an American Robin. You can listen to its distinctive song here:
Listen to the robin: https://xeno-canto.org/12345/download
Another recording: https://xeno-canto.org/67890/download
The American Robin is known for its cheerful song that signals the arrival of spring.
"""
result_2 = await parse_agent_response(
raw_response=test_response_2,
provider="openai",
api_key="test-key",
model="gpt-4o-mini"
)
print("Input length:", len(test_response_2))
print("Output length:", len(result_2))
print("Contains audio links:", "[Listen to recording" in result_2)
print("Contains xeno-canto links:", "xeno-canto.org" in result_2)
print("✅ Test 2 completed")
# Test Case 3: Response with no media (should return original)
print("\n[TEST 3]: Response with no media")
print("-" * 40)
test_response_3 = """
This appears to be a Blue Jay. Blue Jays are intelligent birds known for their problem-solving abilities and distinctive calls.
They are commonly found in North American forests and suburban areas.
"""
result_3 = await parse_agent_response(
raw_response=test_response_3,
provider="openai",
api_key="test-key",
model="gpt-4o-mini"
)
print("Input length:", len(test_response_3))
print("Output length:", len(result_3))
print("Output matches input:", result_3.strip() == test_response_3.strip())
print("✅ Test 3 completed")
# Test Case 4: Response with mixed URLs (images and audio)
print("\n[TEST 4]: Response with mixed media")
print("-" * 40)
test_response_4 = """
This is definitely a Scarlet Tanager. Here are some photos and recordings:
Photo: https://example.com/tanager.jpg
Another photo: https://example.com/tanager2.png
Song recording: https://xeno-canto.org/11111/download
Call recording: https://example.com/tanager.mp3
Scarlet Tanagers are known for their striking red plumage.
"""
result_4 = await parse_agent_response(
raw_response=test_response_4,
provider="openai",
api_key="test-key",
model="gpt-4o-mini"
)
print("Input length:", len(test_response_4))
print("Output length:", len(result_4))
print("Contains image section:", "### Images" in result_4)
print("Contains audio section:", "### Audio Recordings" in result_4)
print("✅ Test 4 completed")
print("\n" + "=" * 60)
print("✅ ALL STRUCTURED OUTPUT TESTS PASSED")
print("=" * 60)
if __name__ == "__main__":
asyncio.run(test_structured_output())
|