kosong

import asyncio import os from pydantic import BaseModel import kosong from kosong import StepResult from kosong.chat_provider.kimi import Kimi from kosong.message import Message from kosong.tooling import CallableTool2, ToolOk, ToolReturnValue from kosong.tooling.simple import SimpleToolset class AddToolParams(BaseModel): a: int b: int class AddTool(CallableTool2[AddToolParams]): name: str = "add" description: str = "Add two integers." params: type[AddToolParams] = AddToolParams async def __call__(self, params: AddToolParams) -> ToolReturnValue: return ToolOk(output=str(params.a + params.b)) async def main() -> None: # Initialize a Kimi chat provider using environment variables for API key kimi = Kimi( base_url=os.environ.get("KIMI_BASE_URL", "https://api.moonshot.ai/v1"), api_key=os.environ.get("KIMI_API_KEY", "your_kimi_api_key_here"), # Replace with your Kimi API key or set KIMI_API_KEY env var model="kimi-k2-turbo-preview", ) # Create a toolset and add the AddTool toolset = SimpleToolset() toolset += AddTool() # Define message history history = [ Message(role="user", content="Please add 2 and 3 with the add tool."), ] # Run an agent step print("Running agent step...") result: StepResult = await kosong.step( chat_provider=kimi, system_prompt="You are a precise math tutor.", toolset=toolset, history=history, ) # Print the generated message and awaited tool results print(f"Generated message: {result.message}") print(f"Tool results: {await result.tool_results()}") if __name__ == "__main__": asyncio.run(main())