40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import asyncio
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
# from pydantic import BaseModel
|
|
from pydantic_ai import Agent
|
|
from pydantic_ai.settings import ModelSettings
|
|
|
|
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,
|
|
system_prompt=config.prompts.system,
|
|
instructions=config.prompts.instructions)
|
|
settings = ModelSettings(
|
|
**config.generation.model_dump()
|
|
)
|
|
|
|
user_prompt = config.prompts.user.format(task="Повтори сообщение")
|
|
result = await agent.run(user_prompt,
|
|
model_settings=settings)
|
|
print(result.output)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |