arpakitlib 1.8.327__py3-none-any.whl → 1.8.328__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.328.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arpakitlib
-Version: 1.8.327
+Version: 1.8.328
 Summary: arpakitlib
 License-Expression: Apache-2.0
 License-File: LICENSE

{arpakitlib-1.8.327.dist-info → arpakitlib-1.8.328.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
@@ -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.328.dist-info/METADATA,sha256=89Z-J3Lhic6uYoQgkP8PN0m5uTDxjwlUMK6Y6UaL24A,3804
+arpakitlib-1.8.328.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+arpakitlib-1.8.328.dist-info/entry_points.txt,sha256=36xqR3PJFT2kuwjkM_EqoIy0qFUDPKSm_mJaI7emewE,87
+arpakitlib-1.8.328.dist-info/licenses/LICENSE,sha256=GPEDQMam2r7FSTYqM1mm7aKnxLaWcBotH7UvQtea-ec,11355
+arpakitlib-1.8.328.dist-info/RECORD,,