arpakitlib 1.8.327__tar.gz → 1.8.332__tar.gz

{arpakitlib-1.8.327 → arpakitlib-1.8.332}/PKG-INFO

@@ -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.332/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 → arpakitlib-1.8.332}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "arpakitlib"
-version = "1.8.327"
+version = "1.8.332"
 description = "arpakitlib"
 authors = [
     { name = "arpakit_company", email = "support@arpakit.com" }
@@ -21,7 +21,6 @@ classifiers = [
     "Topic :: Software Development :: Libraries"
 ]
 dependencies = [
-    "aiogram (>=3.17.0,<4.0.0)",
     "sqlalchemy (>=2.0.37,<3.0.0)",
     "poetry (>=2.0.1,<3.0.0)",
     "pydantic-settings (>=2.7.1,<3.0.0)",
@@ -52,7 +51,6 @@ dependencies = [
     "aiosmtplib (>=4.0.0,<5.0.0)",
     "aiohttp (>=3.11.16,<4.0.0)",
     "aiosqlite (>=0.21.0,<0.22.0)",
-    "pytelegrambotapi (>=4.27.0,<5.0.0)",
     "xlsxwriter (>=3.2.5,<4.0.0)",
     "requests-ntlm (>=1.3.0,<2.0.0)",
     "qrcode (>=8.2,<9.0)",
@@ -64,6 +62,8 @@ dependencies = [
     "redis (>=6.4.0,<7.0.0)",
     "pytest (>=8.4.2,<9.0.0)",
     "poetry-plugin-export (>=1.9.0)",
+    "aiogram (>=3.22.0,<4.0.0)",
+    "pytelegrambotapi (>=4.29.1,<5.0.0)",
 ]
 
 
@@ -92,3 +92,11 @@ arpakitlib = "arpakitlib.ar_arpakitlib_cli_util:execute_arpakitlib_cli"
 [build-system]
 requires = ["poetry-core>=2.0.0,<3.0.0"]
 build-backend = "poetry.core.masonry.api"
+
+
+[dependency-groups]
+full = [
+    "openai (>=2.6.1,<3.0.0)",
+    "asyncssh (>=2.21.1,<3.0.0)",
+    "paramiko (>=4.0.0,<5.0.0)"
+]