Commited backup

fajox1/famods/grokai.py

@@ -0,0 +1,97 @@
+#   █▀▀ ▄▀█   █▀▄▀█ █▀█ █▀▄ █▀
+#   █▀░ █▀█   █░▀░█ █▄█ █▄▀ ▄█
+
+#   https://t.me/famods
+
+# 🔒    Licensed under the GNU AGPLv3
+# 🌐 https://www.gnu.org/licenses/agpl-3.0.html
+
+# ---------------------------------------------------------------------------------
+# Name: GrokAI
+# Description: Взаимодействие с Grok AI
+# meta developer: @FAmods
+# meta banner: https://github.com/FajoX1/FAmods/blob/main/assets/banners/grokai.png?raw=true
+# requires: aiohttp openai
+# ---------------------------------------------------------------------------------
+
+import asyncio
+import logging
+
+from openai import OpenAI
+
+from .. import loader, utils
+
+logger = logging.getLogger(__name__)
+
+@loader.tds
+class GrokAI(loader.Module):
+    """Взаимодействие с Grok AI"""
+
+    strings = {
+        "name": "GrokAI",
+
+        "no_args": "<emoji document_id=5854929766146118183>❌</emoji> <b>Нужно </b><code>{}{} {}</code>",
+        "no_token": "<emoji document_id=5854929766146118183>❌</emoji> <b>Нету токена! Вставь его в </b><code>{}cfg grokai</code>",
+
+        "asking_grok": "<emoji document_id=5325787248363314644>🔄</emoji> <b>Спрашиваю Grok...</b>",
+
+        "answer": """<emoji document_id=5355148941878900494>🌐</emoji> <b>Ответ:</b> {answer}
+
+<emoji document_id=5785419053354979106>❔</emoji> <b>Вопрос:</b> {question}""",
+    }
+
+    def __init__(self):
+        self.config = loader.ModuleConfig(
+            loader.ConfigValue(
+                "api_key",
+                None,
+                lambda: "Токен GrokAI. Получить токен: https://console.x.ai",
+                validator=loader.validators.Hidden(loader.validators.String())
+            ),
+            loader.ConfigValue(
+                "model",
+                "grok-beta",
+                lambda: "Модель Grok AI",
+            ),
+        )
+
+    async def client_ready(self, client, db):
+        self.db = db
+        self._client = client
+
+    @loader.command()
+    async def grok(self, message):
+        """Задать вопрос к Grok"""
+        q = utils.get_args_raw(message)
+        if not q:
+            return await utils.answer(message, self.strings["no_args"].format(self.get_prefix(), "grok", "[вопрос]"))
+
+        if not self.config['api_key']:
+            return await utils.answer(message, self.strings["no_token"].format(self.get_prefix()))
+
+        await utils.answer(message, self.strings['asking_grok'])
+
+        # Не тупите, ЭТО НЕ CHATGPT, это ГРОК от илона маска.
+        # В документации для грока использовалась либа опенаи, так что меня заставили её юзануть ><
+
+        client = OpenAI(
+            api_key=self.config['api_key'],
+            base_url="https://api.x.ai/v1" # вот теперь это точно грок
+        )
+
+        chat_completion = client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": q,
+                }
+            ],
+            model=self.config['model'],
+        )
+
+        return await utils.answer(
+            message,
+            self.strings['answer'].format(
+                question=q, 
+                answer=chat_completion.choices[0].message.content)
+            )