import asyncio
from SimpleLLMFunc import OpenAICompatible, llm_chat
from SimpleLLMFunc.hooks import is_response_yield, is_event_yield

models = OpenAICompatible.load_from_json_file("provider.json")
llm = models["openrouter"]["openai/gpt-4o"]


@llm_chat(llm_interface=llm, stream=True)
async def chat(message: str, history: list | None = None):
    """
    你是一个简洁、有帮助的助手。
    记住对话中之前的上下文。
    """
    pass

async def main():
    history = []

    # 第 1 轮
    print("用户: Python 是什么？")
    print("助手: ", end="")
    async for output in chat("Python 是什么？", history):
        if is_response_yield(output):
            print(output.response, end="")
            history = output.messages
    print("\n")

    # 第 2 轮——Agent 记得第 1 轮的内容
    print("用户: 它的主要特性是什么？")
    print("助手: ", end="")
    async for output in chat("它的主要特性是什么？", history):
        if is_response_yield(output):
            print(output.response, end="")
            history = output.messages
    print()


asyncio.run(main())

from SimpleLLMFunc.hooks import (
    LLMChunkArriveEvent,
    ToolCallStartEvent,
    ToolCallEndEvent,
    ReactEndEvent,
)


async def run_turn(message: str, history: list) -> list:
    async for output in chat(message, history):
        if is_event_yield(output):
            event = output.event
            if isinstance(event, LLMChunkArriveEvent):
                print(event.accumulated_content, end="", flush=True)
            elif isinstance(event, ToolCallStartEvent):
                print(f"\n[调用 {event.tool_name}...]")
            elif isinstance(event, ToolCallEndEvent):
                print(f"[完成: {event.tool_name}]")
            elif isinstance(event, ReactEndEvent):
                return event.final_messages
    return history

@llm_chat(llm_interface=llm, stream=False)
async def simple_chat(message: str, history: list | None = None):
    """一个有帮助的助手。"""
    pass


async def main():
    history = []
    async for output in simple_chat("你好!", history):
        if is_response_yield(output):
            print(output.response)
            history = output.messages


asyncio.run(main())