From 4856114a52ba4501a9d613defa38c75bc0432f18 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=92=D0=B8=D0=BA=D1=82=D0=BE=D1=80?=
 <78488229+viktor138irk@users.noreply.github.com>
Date: Mon, 11 May 2026 22:22:55 +0900
Subject: [PATCH] =?UTF-8?q?=D0=94=D0=BE=D0=B1=D0=B0=D0=B2=D0=BB=D0=B5?=
 =?UTF-8?q?=D0=BD=20automatic=20model=20routing?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/openai_client.py | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/backend/openai_client.py b/backend/openai_client.py
index 8a5910e..ce305fc 100644
--- a/backend/openai_client.py
+++ b/backend/openai_client.py
@@ -1,6 +1,7 @@
 from openai import OpenAI
 
-from backend.config_store import get_openai_key, get_openai_model
+from backend.config_store import get_openai_key
+from backend.model_router import choose_model
 
 
 class OpenAIConfigurationError(Exception):
@@ -18,12 +19,16 @@ def _client() -> OpenAI:
     return OpenAI(api_key=api_key)
 
 
-async def ask_ai(prompt: str):
+async def ask_ai(prompt: str, task_type: str | None = None):
     client = _client()
+    model = choose_model(task_type=task_type, prompt=prompt)
 
     response = client.responses.create(
-        model=get_openai_model(),
+        model=model,
         input=prompt
     )
 
-    return response.output_text
+    return {
+        'model': model,
+        'answer': response.output_text
+    }