feat(solaris): moving to service system to support multi-chat

src/bot/modules/solaris/agents/respond.py

@@ -1,18 +1,31 @@
 import json
-from dataclasses import asdict
 
 from google import genai
+from google.genai import chats, types
 
-from ..content_configs import generate_respond_config
+from utils.config import dconfig
+
+from ..constants import SAFETY_SETTINGS
 from ..structures import InputMessage, OutputMessage
 
-RESPOND_MODEL = "gemini-2.5-flash"
-
 
 class RespondAgent:
-    def __init__(self, client: genai.client.AsyncClient, prompt: str) -> None:
-        self.chat = client.chats.create(
-            model=RESPOND_MODEL, config=generate_respond_config(prompt=prompt)
+    chat: chats.AsyncChat
+
+    def __init__(self, client: genai.client.AsyncClient) -> None:
+        self.client = client
+
+    async def load_chat(self, history: list[types.Content], system_prompt: str):
+        self.chat = self.client.chats.create(
+            model=(await dconfig()).models.respond_model,
+            config=types.GenerateContentConfig(
+                system_instruction=system_prompt,
+                thinking_config=types.ThinkingConfig(thinking_budget=0),
+                response_mime_type="application/json",
+                response_schema=list[OutputMessage],
+                safety_settings=SAFETY_SETTINGS,
+            ),
+            history=history,
         )
 
     async def send_messages(self, messages: list[InputMessage]) -> list[OutputMessage]:

src/bot/modules/solaris/agents/review.py

@@ -2,10 +2,10 @@ import json
 
 from google import genai
 
-from ..content_configs import generate_review_config
+from ..constants import generate_review_config
 from ..structures import InputMessage
 
-REVIEW_MODEL = ("gemini-2.5-flash-lite-preview-06-17",)  # надо будет гемму
+REVIEW_MODEL = "gemini-2.5-flash-lite-preview-06-17"  # надо будет гемму
 
 
 class ReviewAgent:

src/bot/modules/solaris/agents/tts.py

@@ -4,7 +4,7 @@ from google import genai
 from google.genai import types
 from pydub import AudioSegment
 
-from ..content_configs import generate_tts_config
+from ..constants import SAFETY_SETTINGS
 
 TTS_MODEL = "gemini-2.5-flash-preview-tts"
 
@@ -12,18 +12,34 @@ TTS_MODEL = "gemini-2.5-flash-preview-tts"
 class TTSAgent:
     def __init__(self, client: genai.client.AsyncClient) -> None:
         self.client = client
-        self.content_config = generate_tts_config()
+
+        self.content_config = types.GenerateContentConfig(
+            response_modalities=[types.Modality.AUDIO],
+            speech_config=types.SpeechConfig(
+                voice_config=types.VoiceConfig(
+                    prebuilt_voice_config=types.PrebuiltVoiceConfig(
+                        voice_name="Kore",
+                    )
+                )
+            ),
+            safety_settings=SAFETY_SETTINGS,
+        )
 
     async def generate(self, text: str):
         response = await self.client.models.generate_content(
             model=TTS_MODEL, contents=text, config=self.content_config
         )
+
         data = response.candidates[0].content.parts[0].inline_data.data
         pcm_io = io.BytesIO(data)
+        pcm_io.seek(0)
+
         audio = AudioSegment(
             pcm_io.read(), sample_width=2, frame_rate=24000, channels=1
         )
+
         ogg_io = io.BytesIO()
         audio.export(ogg_io, format="ogg", codec="libopus")
         ogg_bytes = ogg_io.getvalue()
+
         return ogg_bytes

src/bot/modules/solaris/client.py

@@ -4,9 +4,8 @@ from .agents import BuildAgent
 
 
 class SolarisClient:
-    def __init__(self, api_key: str) -> None:
-        client = genai.Client(api_key=api_key).aio
-        self.builder = BuildAgent(client=client)
+    def __init__(self, gemini_client: genai.client.AsyncClient) -> None:
+        self.builder = BuildAgent(client=gemini_client)
 
     async def parse_user_data(self, some_data_idk):
         self.reviewer, self.responder = await self.builder.build(

src/bot/modules/solaris/constants.py

@@ -0,0 +1,28 @@
+from google.genai import types
+
+from .structures import OutputMessage
+
+SAFETY_SETTINGS = [
+    types.SafetySetting(category=category, threshold=types.HarmBlockThreshold.OFF)
+    for category in types.HarmCategory
+]
+
+
+def generate_respond_config(prompt: str) -> types.GenerateContentConfig:
+    return
+
+
+def generate_review_config(prompt: str) -> types.GenerateContentConfig:
+    return types.GenerateContentConfig(
+        system_instruction=prompt,
+        thinking_config=types.ThinkingConfig(thinking_budget=0),
+        response_mime_type="application/json",
+        response_schema=list[int],
+        safety_settings=SAFETY_SETTINGS,
+    )
+
+
+# MESSAGE_CONTENT_CONFIG = types.GenerateContentConfig(
+#     response_mime_type="application/json",  # возможно можно скипнуть это если мы используем response_schema, надо проверить
+#     response_schema=list[OutputMessage],
+# )

src/bot/modules/solaris/content_configs.py

@@ -1,51 +0,0 @@
-from google.genai import types
-
-from .structures import OutputMessage
-
-safety_settings = [
-    types.SafetySetting(category=category, threshold=types.HarmBlockThreshold.OFF)
-    for category in types.HarmCategory
-]
-
-
-def generate_respond_config(prompt: str) -> types.GenerateContentConfig:
-    return types.GenerateContentConfig(
-        system_instruction=prompt,
-        thinking_config=types.ThinkingConfig(thinking_budget=0),
-        response_mime_type="application/json",
-        response_schema=list[
-            OutputMessage
-        ],  # ты уверен что там json надо? мне просто каж что судя по всему вот эта нам
-        safety_settings=safety_settings,
-    )
-
-
-def generate_review_config(prompt: str) -> types.GenerateContentConfig:
-    return types.GenerateContentConfig(
-        system_instruction=prompt,
-        thinking_config=types.ThinkingConfig(thinking_budget=0),
-        response_mime_type="application/json",
-        response_schema=list[int],
-        safety_settings=safety_settings,
-    )
-
-
-# MESSAGE_CONTENT_CONFIG = types.GenerateContentConfig(
-#     response_mime_type="application/json",  # возможно можно скипнуть это если мы используем response_schema, надо проверить
-#     response_schema=list[OutputMessage],
-# )
-
-
-# можно было и константу оставить но хезе некрасиво чет
-def generate_tts_config() -> types.GenerateContentConfig:
-    return types.GenerateContentConfig(
-        response_modalities=[types.Modality.AUDIO],
-        speech_config=types.SpeechConfig(
-            voice_config=types.VoiceConfig(
-                prebuilt_voice_config=types.PrebuiltVoiceConfig(
-                    voice_name="Kore",
-                )
-            )
-        ),
-        safety_settings=safety_settings,
-    )

src/bot/modules/solaris/services/respond.py

@@ -0,0 +1,22 @@
+import json
+
+from google import genai
+from google.genai import chats, types
+
+from utils.config import dconfig
+from utils.logging import console
+
+from ..agents.respond import RespondAgent
+from ..constants import SAFETY_SETTINGS
+from ..structures import InputMessage, OutputMessage
+
+
+class RespondService:
+    def __init__(self, client: genai.client.AsyncClient, chat_id: int) -> None:
+        self.agent = RespondAgent(client)
+        self.chat_id = chat_id
+
+    async def spawn_agent(self):
+        console.print(self.chat_id)
+
+        await self.agent.load_chat(history=[], system_prompt="nya nya")