31 lines
846 B
Python
31 lines
846 B
Python
import asyncio
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
# from pydantic import BaseModel
|
|
from pydantic_ai import Agent
|
|
|
|
from llm.config import load_config
|
|
from llm.factory import LLMModelProvider
|
|
|
|
load_dotenv()
|
|
|
|
|
|
async def main():
|
|
|
|
|
|
config = load_config("config.yml")
|
|
model = await LLMModelProvider.create_from_config(config.llm)
|
|
|
|
|
|
# model = await LLMModelProvider.create_codex(model_name='gpt-5.6-luna')
|
|
# model = LLMModelProvider.create_openai_compatible(model_name="qwen/qwen3-8b",
|
|
# api_key=os.getenv("API_KEY"),
|
|
# base_url=os.getenv("BASE_URL"))
|
|
agent = Agent(model=model)
|
|
result = await agent.run('Привет, дай мне ответ')
|
|
print(result.output)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |