comp-ai-task 0.1.0__tar.gz

comp_ai_task-0.1.0/PKG-INFO

@@ -0,0 +1,48 @@
+Metadata-Version: 2.5
+Name: comp-ai-task
+Version: 0.1.0
+Summary: Fast AI calls using task['query'] syntax in Python
+Project-URL: Homepage, https://github.com/yourusername/comp-ai
+Author: Marat
+License: MIT
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+
+# comp_ai
+
+Удобная библиотека для мгновенных запросов к ИИ прямо в Python IDLE или скриптах.
+
+## Установка (после публикации на PyPI)
+
+```bash
+pip install comp-ai-task
+```
+
+## Использование в Python IDLE
+
+```python
+from comp_ai import *
+
+# Отправка запроса:
+task["Напиши стих про космос"]
+
+# Или как функция:
+task("Объясни теорию относительности за 2 предложения")
+```
+
+## Настройка API-ключа
+
+При первом обращении библиотека сама попросит ввести API-ключ Gemini и сохранит его локально.
+
+Также ключ можно указать:
+1. В коде:
+   ```python
+   task.set_key("AIzaSy...")
+   ```
+2. Через переменную окружения:
+   ```bash
+   set GEMINI_API_KEY=AIzaSy...
+   ```

comp_ai_task-0.1.0/README.md

@@ -0,0 +1,35 @@
+# comp_ai
+
+Удобная библиотека для мгновенных запросов к ИИ прямо в Python IDLE или скриптах.
+
+## Установка (после публикации на PyPI)
+
+```bash
+pip install comp-ai-task
+```
+
+## Использование в Python IDLE
+
+```python
+from comp_ai import *
+
+# Отправка запроса:
+task["Напиши стих про космос"]
+
+# Или как функция:
+task("Объясни теорию относительности за 2 предложения")
+```
+
+## Настройка API-ключа
+
+При первом обращении библиотека сама попросит ввести API-ключ Gemini и сохранит его локально.
+
+Также ключ можно указать:
+1. В коде:
+   ```python
+   task.set_key("AIzaSy...")
+   ```
+2. Через переменную окружения:
+   ```bash
+   set GEMINI_API_KEY=AIzaSy...
+   ```

comp_ai_task-0.1.0/comp_ai/__init__.py

@@ -0,0 +1,147 @@
+# -*- coding: utf-8 -*-
+"""
+comp_ai - модуль для быстрых запросов к ИИ в Python IDLE и скриптах.
+"""
+
+import os
+import sys
+import json
+import urllib.request
+import urllib.error
+from pathlib import Path
+from typing import Optional
+
+CONFIG_FILE = Path.home() / ".comp_ai_key"
+
+class AITask:
+    def __init__(self, model: str = "gemini-3.6-flash"):
+        self.model = model
+        self._api_key: Optional[str] = None
+        self._load_key()
+
+    def _load_key(self):
+        """Загрузка ключа из переменных окружения или файла"""
+        env_key = os.getenv("GEMINI_API_KEY")
+        if env_key:
+            self._api_key = env_key.strip()
+            return
+
+        if CONFIG_FILE.exists():
+            try:
+                cached_key = CONFIG_FILE.read_text(encoding="utf-8").strip()
+                if cached_key:
+                    self._api_key = cached_key
+            except Exception:
+                pass
+
+    def set_key(self, api_key: str, save_locally: bool = True):
+        """Установка ключа API"""
+        self._api_key = api_key.strip()
+        if save_locally:
+            try:
+                CONFIG_FILE.write_text(self._api_key, encoding="utf-8")
+            except Exception:
+                pass
+        print("[comp_ai] API-ключ сохранен!")
+
+    @property
+    def key(self) -> Optional[str]:
+        return self._api_key
+
+    @key.setter
+    def key(self, value: str):
+        self.set_key(value)
+
+    def _ensure_api_key(self) -> str:
+        """Запрос ключа, если он не найден"""
+        if self._api_key:
+            return self._api_key
+
+        print("\n" + "=" * 50)
+        print(" [comp_ai] API-ключ Gemini не найден!")
+        print(" Получить бесплатный ключ: https://aistudio.google.com/app/apikey")
+        print("=" * 50)
+
+        try:
+            user_input = input(" Введите ваш GEMINI_API_KEY: ").strip()
+        except (EOFError, KeyboardInterrupt):
+            user_input = ""
+
+        if not user_input:
+            raise ValueError(
+                "API-ключ не указан! Задайте его через task.set_key('...') или переменную GEMINI_API_KEY"
+            )
+
+        self.set_key(user_input, save_locally=True)
+        return self._api_key
+
+    def query(self, prompt: str) -> str:
+        """Вызов Gemini API"""
+        api_key = self._ensure_api_key()
+        
+        # Если модель не сработала, пробуем fallback-модели
+        fallback_models = [self.model, "gemini-2.0-flash", "gemini-1.5-flash"]
+        # Убираем дубликаты сохраняя порядок
+        models_to_try = []
+        for m in fallback_models:
+            if m not in models_to_try:
+                models_to_try.append(m)
+
+        last_error = ""
+        for mod in models_to_try:
+            url = f"https://generativelanguage.googleapis.com/v1beta/models/{mod}:generateContent?key={api_key}"
+            headers = {"Content-Type": "application/json"}
+            payload = {"contents": [{"parts": [{"text": str(prompt)}]}]}
+
+            req = urllib.request.Request(
+                url,
+                data=json.dumps(payload).encode("utf-8"),
+                headers=headers,
+                method="POST"
+            )
+
+            try:
+                with urllib.request.urlopen(req) as resp:
+                    resp_data = json.loads(resp.read().decode("utf-8"))
+                    candidates = resp_data.get("candidates", [])
+                    if not candidates:
+                        text_res = "[Пустой ответ от модели]"
+                    else:
+                        parts = candidates[0].get("content", {}).get("parts", [])
+                        text_res = "".join(p.get("text", "") for p in parts)
+
+                    # Запоминаем рабочую модель
+                    self.model = mod
+                    print(text_res)
+                    return text_res
+
+            except urllib.error.HTTPError as e:
+                error_body = e.read().decode("utf-8", errors="ignore")
+                last_error = f"[Ошибка API Gemini {e.code}]: {error_body}"
+                if e.code == 404:
+                    # Пробуем следующую модель из списка
+                    continue
+                else:
+                    print(last_error, file=sys.stderr)
+                    return last_error
+            except Exception as e:
+                last_error = f"[Ошибка запроса]: {e}"
+                print(last_error, file=sys.stderr)
+                return last_error
+
+        print(last_error, file=sys.stderr)
+        return last_error
+
+    def __getitem__(self, prompt: str) -> str:
+        return self.query(prompt)
+
+    def __call__(self, prompt: str) -> str:
+        return self.query(prompt)
+
+    def __repr__(self) -> str:
+        status = "готов" if self._api_key else "ожидает ключ"
+        return f"<comp_ai.task (модель: {self.model}, статус: {status})>"
+
+
+task = AITask()
+__all__ = ["task", "AITask"]

comp_ai_task-0.1.0/pyproject.toml

@@ -0,0 +1,32 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "comp-ai-task"
+version = "0.1.0"
+description = "Fast AI calls using task['query'] syntax in Python"
+readme = "README.md"
+requires-python = ">=3.8"
+license = { text = "MIT" }
+authors = [
+    { name = "Marat" }
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[tool.hatch.build.targets.wheel]
+packages = ["comp_ai"]
+
+[tool.hatch.build.targets.sdist]
+include = [
+    "/comp_ai",
+    "/README.md",
+    "/pyproject.toml"
+]
+
+[project.urls]
+"Homepage" = "https://github.com/yourusername/comp-ai"