test_format.py

import asyncio
import os
import sys

sys.path.append(os.path.dirname(os.path.abspath(__file__)))

from chatgpt.ChatService import ChatService
from chatgpt.chatFormat import api_messages_to_chat

async def main():
    token = None
    try:
        with open('data/token.txt', 'r') as f:
            tokens = [line.strip() for line in f if line.strip()]
            if tokens:
                token = tokens[-1]
    except Exception:
        pass

    cs = ChatService(token if token else "sk-123")
    api_messages = [{"role": "user", "content": "Draw me a small simple icon of an apple."}]
    cs.api_messages = api_messages
    await cs.set_dynamic_data({"model": "gpt-4o"})
    try:
        res = await api_messages_to_chat(cs, api_messages, False)
        print("Success:", res)
    except Exception as e:
        import traceback
        traceback.print_exc()

asyncio.run(main())