feat(solaris): navalil dobrenko
This commit is contained in:
@@ -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]:
|
||||
|
||||
Reference in New Issue
Block a user