feat(solaris): navalil dobrenko

This commit is contained in:
shinrei
2025-07-02 22:17:44 +00:00
parent bf779f40d9
commit a4700b441e
6 changed files with 50 additions and 19 deletions

View File

@@ -6,12 +6,13 @@ from google import genai
from ..content_configs import generate_respond_config
from ..structures import InputMessage, OutputMessage
RESPOND_MODEL = "gemini-2.5-flash"
class RespondAgent:
def __init__(self, api_key: str) -> None:
client = genai.Client(api_key=api_key).aio
def __init__(self, client: genai.client.AsyncClient, prompt: str) -> None:
self.chat = client.chats.create(
model="gemini-2.5-flash", config=generate_respond_config(prompt="im gay")
model=RESPOND_MODEL, config=generate_respond_config(prompt=prompt)
)
async def send_messages(self, messages: list[InputMessage]) -> list[OutputMessage]: