From 19f4075e84077e974d3b0c64787803b5907603fd Mon Sep 17 00:00:00 2001
From: Anton Potapov <47938145+Flagro@users.noreply.github.com>
Date: Mon, 14 Oct 2024 14:59:11 +0200
Subject: [PATCH] Fix return type of _get_model_by_name

---
 bot/rp_bot/ai_agent/ai.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/bot/rp_bot/ai_agent/ai.py b/bot/rp_bot/ai_agent/ai.py
index cb908f7..a203582 100644
--- a/bot/rp_bot/ai_agent/ai.py
+++ b/bot/rp_bot/ai_agent/ai.py
@@ -1,6 +1,6 @@
 import tiktoken
 import io
-from typing import AsyncIterator, Literal, Optional
+from typing import AsyncIterator, Literal
 from openai import OpenAI
 from langchain_openai import OpenAI
 from langchain_core.messages import HumanMessage, SystemMessage
@@ -57,7 +57,7 @@ def _get_default_model_name(
 
     def _get_model_by_name(
         self, model_name: str, model_type: Literal["text", "vision", "image_generation"]
-    ) -> Optional[Model]:
+    ) -> Model:
         models_dict = {
             "text": self.ai_config.TextGeneration.Models,
             "vision": self.ai_config.TextGeneration.Models,