import sys
import os
import asyncio
import logging
import traceback

# Add project root to path
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

# Configure logging
logging.basicConfig(level=logging.INFO)

from Util.GeminiClient import GeminiClient

async def main():
    print("=== Testing LingYa LLM via Util.GeminiClient ===")
    
    try:
        # 使用 LingYa 提供商初始化客户端
        client = GeminiClient(provider="LingYa")
        
        messages = [
            {"role": "developer", "content": "你是一个幽默的助手。"},
            {"role": "user", "content": "用一句话形容程序员的生活。"}
        ]
        
        # 使用 gpt-5.2 进行测试，流式响应更稳定
        # 项目默认配置可能是 gemini-3-pro-preview
        # model = Config.LINGYA_API_GEMINI_MODEL
        model = "gpt-5.2" 
        print(f"Requesting with model: {model}")
        print(f"\n[Response]:")
        
        async for content in client.send_chat_completion_stream(messages, model=model):
            print(content, end="", flush=True)
        print("\n")
            
    except BaseException as e:
        print(f"Test failed: {e}")
        traceback.print_exc()

if __name__ == "__main__":
    try:
        asyncio.run(main())
    except KeyboardInterrupt:
        print("\nInterrupted by user")
    except BaseException as e:
        print(f"\nMain failed: {e}")
        traceback.print_exc()