arpakitlib 1.8.327__py3-none-any.whl → 1.8.332__py3-none-any.whl

arpakitlib/ar_openai_api_client_util.py

@@ -0,0 +1,204 @@
+from __future__ import annotations
+
+import asyncio
+import logging
+
+import httpx
+from arpakitlib.ar_base64_util import convert_file_to_base64_string
+from openai import OpenAI, AsyncOpenAI
+from openai.types.chat import ChatCompletion
+
+"""
+https://platform.openai.com/docs/
+"""
+
+
+class EasyOpenAIAPIClient:
+    def __init__(
+            self,
+            *,
+            open_ai: OpenAI,
+            async_open_ai: AsyncOpenAI
+    ):
+        self._logger = logging.getLogger(self.__class__.__name__)
+
+        self.open_ai = open_ai
+        self.async_open_ai = async_open_ai
+
+    @classmethod
+    def create_easily(
+            cls,
+            openai_api_key: str,
+            openai_api_base_url: str | None = "https://api.proxyapi.ru/openai/v1"
+    ) -> EasyOpenAIAPIClient:
+        return EasyOpenAIAPIClient(
+            open_ai=OpenAI(
+                api_key=openai_api_key,
+                base_url=openai_api_base_url,
+                timeout=httpx.Timeout(
+                    timeout=300,  # общий таймаут
+                    connect=5,  # таймаут подключения
+                    read=300,  # чтение ответа
+                    write=60,  # запись запроса
+                    pool=10
+                )
+            ),
+            async_open_ai=AsyncOpenAI(
+                api_key=openai_api_key,
+                base_url=openai_api_base_url,
+                http_client=httpx.AsyncClient(
+                    timeout=httpx.Timeout(
+                        timeout=300,
+                        connect=5,
+                        read=300,
+                        write=60,
+                        pool=10
+                    )
+                )
+            )
+        )
+
+    def check_conn(self):
+        self.open_ai.models.list()
+
+    def is_conn_good(self) -> bool:
+        try:
+            self.check_conn()
+            return True
+        except Exception as e:
+            self._logger.error(e)
+        return False
+
+    async def async_check_conn(self):
+        await self.async_open_ai.models.list()
+
+    async def async_is_conn_good(self) -> bool:
+        try:
+            await self.async_check_conn()
+            return True
+        except Exception as e:
+            self._logger.error(e)
+        return False
+
+    def simple_ask(
+            self,
+            *,
+            prompt: str | None = None,
+            text: str,
+            model: str = "gpt-4o",
+            image_links: str | list[str] | None = None,
+            image_filepaths: str | list[str] | None = None
+    ) -> ChatCompletion:
+        if isinstance(image_links, str):
+            image_links = [image_links]
+        if isinstance(image_filepaths, str):
+            image_filepaths = [image_filepaths]
+
+        messages = []
+
+        if prompt is not None:
+            messages.append({
+                "role": "system",
+                "content": prompt
+            })
+
+        content = [{"type": "text", "text": text}]
+
+        if image_links:
+            for link in image_links:
+                content.append({
+                    "type": "image_url",
+                    "image_url": {"url": link}
+                })
+
+        if image_filepaths:
+            for path in image_filepaths:
+                base64_url = convert_file_to_base64_string(filepath=path, raise_for_error=True)
+                content.append({
+                    "type": "image_url",
+                    "image_url": {"url": base64_url}
+                })
+
+        messages.append({
+            "role": "user",
+            "content": content
+        })
+
+        response: ChatCompletion = self.open_ai.chat.completions.create(
+            model=model,
+            messages=messages,
+            n=1,
+            temperature=0.1,
+            top_p=0.9,
+            max_tokens=1000
+        )
+
+        return response
+
+    async def async_simple_ask(
+            self,
+            *,
+            prompt: str | None = None,
+            text: str,
+            model: str = "gpt-4o",
+            image_links: str | list[str] | None = None,
+            image_filepaths: str | list[str] | None = None
+    ) -> ChatCompletion:
+        if isinstance(image_links, str):
+            image_links = [image_links]
+        if isinstance(image_filepaths, str):
+            image_filepaths = [image_filepaths]
+
+        messages = []
+
+        if prompt is not None:
+            messages.append({
+                "role": "system",
+                "content": prompt
+            })
+
+        content = [{"type": "text", "text": text}]
+
+        if image_links:
+            for link in image_links:
+                content.append({
+                    "type": "image_url",
+                    "image_url": {"url": link}
+                })
+
+        if image_filepaths:
+            for path in image_filepaths:
+                base64_url = convert_file_to_base64_string(filepath=path, raise_for_error=True)
+                content.append({
+                    "type": "image_url",
+                    "image_url": {"url": base64_url}
+                })
+
+        messages.append({
+            "role": "user",
+            "content": content
+        })
+
+        response: ChatCompletion = await self.async_open_ai.chat.completions.create(
+            model=model,
+            messages=messages,
+            n=1,
+            temperature=0.1,
+            top_p=0.9,
+            max_tokens=1000
+        )
+
+        return response
+
+
+def __example():
+    pass
+
+
+async def __async_example():
+    pass
+
+
+if __name__ == '__main__':
+    __example()
+    asyncio.run(__async_example())

{arpakitlib-1.8.327.dist-info → arpakitlib-1.8.332.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arpakitlib
-Version: 1.8.327
+Version: 1.8.332
 Summary: arpakitlib
 License-Expression: Apache-2.0
 License-File: LICENSE
@@ -17,7 +17,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Operating System :: OS Independent
 Classifier: Topic :: Software Development :: Libraries
 Requires-Dist: aio-pika (>=9.5.4,<10.0.0)
-Requires-Dist: aiogram (>=3.17.0,<4.0.0)
+Requires-Dist: aiogram (>=3.22.0,<4.0.0)
 Requires-Dist: aiohttp (>=3.11.16,<4.0.0)
 Requires-Dist: aiohttp-socks (>=0.10.1,<0.11.0)
 Requires-Dist: aiosmtplib (>=4.0.0,<5.0.0)
@@ -45,7 +45,7 @@ Requires-Dist: pydantic (>=2.10.5,<3.0.0)
 Requires-Dist: pydantic-settings (>=2.7.1,<3.0.0)
 Requires-Dist: pyjwt (>=2.10.1,<3.0.0)
 Requires-Dist: pymongo (>=4.10.1,<5.0.0)
-Requires-Dist: pytelegrambotapi (>=4.27.0,<5.0.0)
+Requires-Dist: pytelegrambotapi (>=4.29.1,<5.0.0)
 Requires-Dist: pytest (>=8.4.2,<9.0.0)
 Requires-Dist: pytz (>=2024.2,<2025.0)
 Requires-Dist: qrcode (>=8.2,<9.0)

{arpakitlib-1.8.327.dist-info → arpakitlib-1.8.332.dist-info}/RECORD

@@ -408,6 +408,7 @@ arpakitlib/ar_log_async_func_if_error_util.py,sha256=8qtcxRk6OqfwVsGE2wCpuIY7fju
 arpakitlib/ar_logging_util.py,sha256=8TdaCWeSk4_GAmuWFO49l8JAVfLJCqUxgSpTYvlWvsk,1308
 arpakitlib/ar_mongodb_util.py,sha256=orRb81o9CzD9kkWbbNQCbSMTl2xLcctUDdx_FxirKww,3411
 arpakitlib/ar_need_type_util.py,sha256=XmY1kswz8j9oo5f9CxRu0_zgfvxWrXPYKOj6MM04sGk,2604
+arpakitlib/ar_openai_api_client_util.py,sha256=dIPHNrDRkclRa__qPj7glkv232C97tzAAFEFuXtvV1E,5569
 arpakitlib/ar_parse_command_util.py,sha256=1WTdQoWVshoDZ1jDaKeTzajfqaYHP3FNO0-REyo1aMY,3003
 arpakitlib/ar_postgresql_util.py,sha256=1AuLjEaa1Lg4pzn-ukCVnDi35Eg1k91APRTqZhIJAdo,945
 arpakitlib/ar_pydantic_schema_from_sqlalchemy_model_util.py,sha256=SddmsEdRdVbmdZ5aSuwy6DfGYmYx_JO6lKzuEJeBAV8,10553
@@ -417,7 +418,7 @@ arpakitlib/ar_really_validate_email_util.py,sha256=HBfhyiDB3INI6Iq6hR2WOMKA5wVWW
 arpakitlib/ar_really_validate_url_util.py,sha256=aaSPVMbz2DSqlC2yk2g44-kTIiHlITfJwIG97L-Y93U,1309
 arpakitlib/ar_retry_func_util.py,sha256=LB4FJRsu2cssnPw6X8bCEcaGpQsXhkLkgeU37w1t9fU,2250
 arpakitlib/ar_run_cmd_util.py,sha256=D_rPavKMmWkQtwvZFz-Io5Ak8eSODHkcFeLPzNVC68g,1072
-arpakitlib/ar_safe_func.py,sha256=JQNaM3q4Z6nUE8bDfzXNBGkWSXm0PZ-GMMvu6UWUIYk,2400
+arpakitlib/ar_safe_func_util.py,sha256=JQNaM3q4Z6nUE8bDfzXNBGkWSXm0PZ-GMMvu6UWUIYk,2400
 arpakitlib/ar_settings_util.py,sha256=FeQQkuVrLVYkFAIg3Wy6ysyTt_sqLTX0REAe60gbM3k,2361
 arpakitlib/ar_sleep_util.py,sha256=ggaj7ML6QK_ADsHMcyu6GUmUpQ_9B9n-SKYH17h-9lM,1045
 arpakitlib/ar_sqladmin_util.py,sha256=SEoaowAPF3lhxPsNjwmOymNJ55Ty9rmzvsDm7gD5Ceo,861
@@ -428,8 +429,8 @@ arpakitlib/ar_str_util.py,sha256=6KlLL-SB8gzK-6gwQEd3zuYbRvtjd9HFpJ9-xHbkH6U,435
 arpakitlib/ar_type_util.py,sha256=Cs_tef-Fc5xeyAF54KgISCsP11NHyzIsglm4S3Xx7iM,4049
 arpakitlib/ar_uppercase_env_keys_util.py,sha256=BsUCJhfchBIav0AE54_tVgYcE4p1JYoWdPGCHWZnROA,2790
 arpakitlib/ar_yookassa_api_client_util.py,sha256=7DL_0GyIOTuSNBHUO3qWxAXMKlBRHjKgcA6ttst8k1A,5265
-arpakitlib-1.8.327.dist-info/METADATA,sha256=-pMxjG5dQPSW-_vsrE8TI3YNBDASDHJ-h3NzbKJT1JA,3804
-arpakitlib-1.8.327.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-arpakitlib-1.8.327.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
-arpakitlib-1.8.327.dist-info/licenses/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
-arpakitlib-1.8.327.dist-info/RECORD,,
+arpakitlib-1.8.332.dist-info/METADATA,sha256=EohLsrJDv0HiRnX78HHtdWjNe68iHoddMF-yXxDarQU,3804
+arpakitlib-1.8.332.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+arpakitlib-1.8.332.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.332.dist-info/licenses/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.332.dist-info/RECORD,,