#!/usr/bin/env python3
"""
Test OllamaAdapter XML parsing with real malformed examples.
"""

import asyncio
import sys
sys.path.insert(0, '/home/serversdown/project-lyra/cortex')

from autonomy.tools.adapters.ollama_adapter import OllamaAdapter


async def test_parser():
    adapter = OllamaAdapter()

    # Test cases with actual malformed XML we've seen
    test_cases = [
        {
            "name": "Malformed closing tag 1",
            "xml": """<tool_call>
  <name>execute_code</name>
  <arguments>
    <language>python</language>
    <code>print(50 / 2)</code>
    <reason>To calculate the result of dividing 50 by 2.</reason>
  </arguments>
</olith>"""
        },
        {
            "name": "Malformed closing tag 2",
            "xml": """<tool_call>
  <name>execute_code</name>
  <arguments>
    <language>python</language>
    <code>print(60 / 4)</code>
  </arguments>
  <reason>To calculate 60 divided by 4 using Python.</reason>
</LTRB"""
        },
        {
            "name": "Partial XML (missing opening)",
            "xml": """<arguments>
    <language>python</language>
    <code>result = 35 / 7; result</code>
  </arguments>
  <reason>To calculate the division of 35 by 7 using Python.</reason>
</tool_call>"""
        },
        {
            "name": "Correct XML",
            "xml": """<tool_call>
  <name>execute_code</name>
  <arguments>
    <language>python</language>
    <code>print(100 / 4)</code>
    <reason>Calculate division</reason>
  </arguments>
</tool_call>"""
        },
        {
            "name": "XML with surrounding text",
            "xml": """Let me help you with that.

<tool_call>
  <name>execute_code</name>
  <arguments>
    <language>python</language>
    <code>print(20 / 4)</code>
    <reason>Calculate the result</reason>
  </arguments>
</tool_call>

The result will be shown above."""
        }
    ]

    print("=" * 80)
    print("Testing OllamaAdapter XML Parsing")
    print("=" * 80)

    for test in test_cases:
        print(f"\nTest: {test['name']}")
        print("-" * 80)
        print(f"Input XML:\n{test['xml'][:200]}{'...' if len(test['xml']) > 200 else ''}")
        print("-" * 80)

        try:
            result = await adapter.parse_response(test['xml'])
            print(f"✅ Parsed successfully!")
            print(f"   Content: {result.get('content', '')[:100]}")
            print(f"   Tool calls found: {len(result.get('tool_calls') or [])}")

            if result.get('tool_calls'):
                for idx, tc in enumerate(result['tool_calls']):
                    print(f"   Tool {idx + 1}: {tc.get('name')} with args: {tc.get('arguments')}")
        except Exception as e:
            print(f"❌ Error: {e}")

        print()


if __name__ == "__main__":
    asyncio.run(test_parser())