#!/usr/bin/env python3
"""
Test script for basic message exchange with LLM.
"""

from config import Config
from llm_client import LLMClient
from conversation import ConversationManager


def test_message_exchange():
    """Test basic message exchange with LLM."""
    print("Testing message exchange with LLM...")
    print("-" * 40)
    
    # Create components
    config = Config()
    llm_client = LLMClient(config)
    conversation_manager = ConversationManager()
    
    # Test connection first
    try:
        success = llm_client.test_connection()
        if not success:
            print("✗ Failed to connect to LM Studio")
            return
        else:
            print("✓ Connected to LM Studio")
    except Exception as e:
        print(f"✗ Error testing connection: {e}")
        return
    
    # Add a test message to conversation
    test_message = "Hello, are you there?"
    conversation_manager.add_user_message(test_message)
    print(f"Sending message: {test_message}")
    
    # Get response from LLM
    try:
        response = llm_client.get_response(conversation_manager.get_history())
        print(f"Received response: {response}")
        
        # Add response to conversation
        conversation_manager.add_assistant_message(response)
        
        print("\n✓ Message exchange successful!")
        
    except Exception as e:
        print(f"✗ Error in message exchange: {e}")


if __name__ == "__main__":
    test_message_exchange()