llm = LiteLlm(
    model=os.getenv("BASE_MODEL"),
    api_base=os.getenv("BASE_URL"),
    api_key=os.getenv("API_KEY"),
)

base_planner = BuiltInPlanner(thinking_config=ThinkingConfig(include_thoughts=True))

agent = Agent(
    name="simple_agent",
    model=llm,
    instruction="你是一个乐于助人的中文助手",
    description="回答用户的python编程问题",
    planner=base_planner
)

session_service = InMemorySessionService()

session = session_service.create_session_sync(
    app_name="adk_project",
    user_id="1",
    session_id="1"
)

runner = Runner(
    app_name="adk_project",
    agent=agent,
    session_service=session_service,
)

if __name__ == '__main__':
    content = types.Content(role="user", parts=[
        types.Part(text="我想要请求一个外部服务，通过http，你有什么好的建议？以'经过以下思考：'作为回答的开头")])
    print("开始调用模型")

    thinking = False
    for event in runner.run(user_id="1", session_id="1", new_message=content,
                            run_config=RunConfig(streaming_mode=StreamingMode.SSE)):
        print(event)