askai-python 0.1.6__py3-none-any.whl → 0.2.1__py3-none-any.whl

ask_ai/base.py

@@ -9,9 +9,30 @@ class Response:
     """
     Unified response object for all easyai requests.
     """
-    def __init__(self, text: str = "", media: Union[ImageObject, AudioObject, None] = None):
-        self.text = text
+    def __init__(self, text: str = "", media: Union[ImageObject, AudioObject, None] = None, **kwargs):
+        self._raw_text = text
         self.media = media
+        self._kwargs = kwargs
+
+    @property
+    def text(self) -> str:
+        text = self._raw_text
+        if self._kwargs.get('strip'):
+            from .utils import strip_tags
+            text = strip_tags(text)
+        if self._kwargs.get('clean'):
+            from .utils import clean_markdown
+            text = clean_markdown(text)
+        if self._kwargs.get('code'):
+            from .utils import extract_code
+            text = extract_code(text)
+        return text
+
+    @property
+    def json(self) -> Union[Dict[str, Any], list]:
+        """Returns parsed JSON. If 'json=True' was passed, this is highly reliable."""
+        from .utils import parse_json
+        return parse_json(self._raw_text)
 
     def __str__(self):
         return self.text
@@ -82,18 +103,70 @@ class BaseProvider:
         # 3. Check for specific output_type override (e.g. user forced image)
         output_type = kwargs.get('output_type')
         
-        # 4. Call Provider Implementation
-        return self._send_request(messages, final_config, output_type)
+        # 4. Extract timeout and retry safely
+        timeout_val = kwargs.get('timeout', final_config.extra.get('timeout', 30))
+        retry_val = kwargs.get('retry', final_config.extra.get('retry', 0))
+        
+        timeout = float(timeout_val) if timeout_val is not None else 30.0
+        retry = int(retry_val) if retry_val is not None else 0
+        
+        final_config.extra['timeout'] = timeout # Pass down to providers
+
+        # 5. Execute with Retry Logic
+        attempts = 0
+        last_error = None
+        while attempts <= retry:
+            try:
+                response = self._send_request(messages, final_config, output_type)
+                # Attach parsing arguments to the response object dynamically
+                response._kwargs.update(kwargs)
+                return response
+            except Exception as e:
+                last_error = e
+                if self._is_retryable_error(e) and attempts < retry:
+                    attempts += 1
+                    delay = 2 ** attempts # 2s, 4s, 8s...
+                    print(f"[ask-ai] Transient error ({e.__class__.__name__}). Retrying in {delay}s... ({attempts}/{retry})")
+                    import time
+                    time.sleep(delay)
+                else:
+                    raise e
+                    
+        # Fallback if loop exits without returning
+        from .exceptions import AskAINetworkError
+        raise AskAINetworkError(f"Request failed after {retry} retries. Last error: {last_error}")
+
+    def _is_retryable_error(self, e: Exception) -> bool:
+        """Dynamically detect rate limits and network timeouts across different provider SDKs."""
+        err_str = str(e).lower()
+        err_cls = e.__class__.__name__.lower()
+        
+        # Explicit ask-ai errors
+        from .exceptions import AskAINetworkError, AskAIRateLimitError
+        if isinstance(e, (AskAINetworkError, AskAIRateLimitError)): return True
+        
+        # Heuristics for underlying provider SDKs (OpenAI, Anthropic, etc.)
+        if "ratelimit" in err_cls or "rate_limit" in err_cls or "429" in err_str: return True
+        if "timeout" in err_cls or "timeout" in err_str: return True
+        if "connection" in err_cls or "connecterror" in err_cls: return True
+        
+        return False
 
     def _merge_configs(self, global_conf: AdvancedConfig, req_conf: AdvancedConfig) -> AdvancedConfig:
         return global_conf.merge(req_conf) # We need to implement merge logic in AdvancedConfig properly or here
 
     def _prepare_messages(self, query: str, config: AdvancedConfig) -> List[Dict[str, str]]:
         messages = []
-        if config.system_message:
-            messages.append({"role": "system", "content": config.system_message})
-        elif self.persona:
-            messages.append({"role": "system", "content": self.persona})
+        
+        system_msg = config.system_message or self.persona or ""
+        
+        # Smart capability: if json=True, enforce JSON system prompt
+        if config.extra.get('json'):
+            json_instruction = "You must respond with valid JSON only, without any markdown formatting or tags."
+            system_msg = f"{system_msg}\n{json_instruction}".strip()
+
+        if system_msg:
+            messages.append({"role": "system", "content": system_msg})
         
         messages.append({"role": "user", "content": query})
         return messages

ask_ai/exceptions.py

@@ -2,14 +2,29 @@ class AskAIError(Exception):
     """Base exception for all ask-ai errors."""
     pass
 
-class APIKeyError(AskAIError):
-    """Raised when API key is missing or invalid."""
+class AskAIConfigError(AskAIError):
+    """Raised when configuration (like API keys) is missing or invalid."""
+    pass
+
+class AskAINetworkError(AskAIError):
+    """Raised when a network connection fails (timeout, disconnect)."""
+    pass
+
+class AskAIRateLimitError(AskAIError):
+    """Raised when the provider rate limits the request (429)."""
     pass
 
 class ProviderError(AskAIError):
-    """Raised when the provider API fails (e.g. 500 error)."""
+    """Raised when the provider API fails (e.g. 500 error or bad request)."""
     pass
 
 class MediaTypeNotSupportedError(AskAIError):
     """Raised when a provider can't handle the requested media type."""
     pass
+
+class AskAIParsingError(AskAIError):
+    """Raised when expect_json is True but the response cannot be parsed."""
+    pass
+
+# Alias for backwards compatibility internally
+APIKeyError = AskAIConfigError

ask_ai/utils.py

@@ -0,0 +1,43 @@
+import re
+import json
+from typing import Dict, Any, Union
+
+def clean_markdown(text: str) -> str:
+    """Removes markdown code blocks like ```json ... ```"""
+    text = text.strip()
+    if text.startswith("```"):
+        # find the first newline
+        first_newline = text.find("\n")
+        if first_newline != -1:
+            text = text[first_newline+1:]
+        if text.endswith("```"):
+            text = text[:-3]
+    return text.strip()
+
+def extract_code(text: str) -> str:
+    """Extracts code blocks from text, ignoring conversational filler."""
+    text = str(text)
+    matches = re.findall(r'```(?:[\w]*\n)?(.*?)```', text, re.DOTALL)
+    if matches:
+        return "\n\n".join(m.strip() for m in matches)
+    return text
+
+def strip_tags(text: str) -> str:
+    """Removes XML/HTML tags, including <think>...</think> blocks."""
+    # First remove <think>...</think> entirely if we don't want the thought process
+    text = re.sub(r'<think>.*?</think>', '', text, flags=re.DOTALL)
+    # Remove any other HTML tags
+    text = re.sub(r'<[^>]+>', '', text)
+    return text.strip()
+
+def parse_json(text: str) -> Union[Dict[str, Any], list]:
+    """Tries to parse JSON from text, employing cleaning if necessary."""
+    try:
+        return json.loads(text)
+    except json.JSONDecodeError:
+        cleaned = clean_markdown(text)
+        try:
+            return json.loads(cleaned)
+        except json.JSONDecodeError as e:
+            from .exceptions import AskAIParsingError
+            raise AskAIParsingError(f"Failed to parse JSON: {e}\nContent snippet: {text[:100]}...")

askai_python-0.2.1.dist-info/METADATA

@@ -0,0 +1,137 @@
+Metadata-Version: 2.4
+Name: askai-python
+Version: 0.2.1
+Summary: AI Made Stupid Simple. Unified API for OpenAI, Groq, Google, and more.
+Author-email: Hossein Ghorbani <hosseingh1068@gmail.com>
+License: MIT
+Project-URL: Homepage, https://hosseinghorbani0.ir/
+Project-URL: Bug Tracker, https://github.com/Hosseinghorbani0/ask-ai/issues
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.0.0
+Requires-Dist: groq
+Requires-Dist: google-generativeai
+Requires-Dist: anthropic
+Requires-Dist: requests
+Requires-Dist: Pillow
+Dynamic: license-file
+
+# askai-python 🚀
+
+**A minimal Python SDK to switch between LLM providers in one line.**  
+No frameworks. No servers. No overengineering.
+
+[![PyPI version](https://img.shields.io/pypi/v/askai-python.svg)](https://pypi.org/project/askai-python/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![GitHub](https://img.shields.io/badge/GitHub-View%20Source-blue.svg)](https://github.com/Hosseinghorbani0/ask-ai)
+
+---
+
+## ⚡ Quick Start (5 seconds)
+
+```bash
+pip install askai-python
+```
+
+```python
+from ask_ai import OpenAI, Groq
+
+# Auto-detects OPENAI_API_KEY from environment
+print(OpenAI().ask("Explain black holes like I'm 5").text)
+
+# Switch provider instantly
+print(Groq().ask("Explain black holes like I'm 5").text)
+```
+
+---
+
+## 🧐 Why ask-ai?
+
+- **One function**: Just call `.ask()`
+- **Multiple providers**: OpenAI, Anthropic, Google Gemini, Groq, Azure, OpenRouter
+- **Zero config**: Keys are pulled from the environment automatically
+- **SDK-first, not a framework**: It stays out of your way.
+
+## 🚫 What this project is NOT
+
+> ❌ Not an AI framework  
+> ❌ Not an API gateway  
+> ❌ Not an agent memory system  
+
+It does one thing perfectly: **Simplifying the API call to LLMs.**
+
+---
+
+## 🛠️ Advanced Usage
+
+### 🧰 Developer QoL Utilities (Auto-Parsing)
+Stop writing Regex to clean up model outputs! `ask-ai` comes with built-in text processing flags:
+
+```python
+from ask_ai import OpenAI
+ai = OpenAI()
+
+# 1. Clean Markdown (Removes ```json and ``` tags)
+# Perfect for extracting raw data from models that wrap everything in markdown
+clean_text = ai.ask("Write JSON", clean=True).text
+
+# 2. Extract Code (Returns ONLY the code block, ignores conversational filler)
+# Great for automation pipelines
+code = ai.ask("Write a python ping script", code=True).text
+
+# 3. Strip Tags (Removes <think> blocks and HTML)
+# Essential for reasoning models like DeepSeek-R1
+answer_only = ai.ask("What is 1+1?", strip=True).text
+
+# 4. Enforce & Parse JSON (Directly returns a Parsed Python Dictionary)
+# Adds JSON instructions to the prompt and safely runs json.loads()
+data_dict = ai.ask("Extract user info", json=True).json
+print(data_dict['name'])
+```
+
+## 🚀 Built-in Resiliency (Retries & Timeouts)
+
+Build reliable apps without writing your own loops. `askai-python` handles rate limits (`429`) and network drops via an internal exponential backoff.
+
+```python
+from ask_ai import OpenAI
+
+ai = OpenAI()
+
+# Automatically retries up to 3 times on transient errors, with a 15-second timeout
+response = ai.ask(
+    "Write a complex python script", 
+    retry=3, 
+    timeout=15 
+)
+```
+
+## 🔌 Supported Providers
+
+| Provider | Class | Capabilities |
+|----------|-------|-------------|
+| **OpenAI** | `OpenAI` | Text, Images (DALL-E), Vision |
+| **Anthropic** | `Anthropic` | Text, Vision (Claude 3.5) |
+| **Google** | `Google` | Text, Images, Video, Audio |
+| **Groq** | `Groq` | Ultra-fast Llama 3 & Mixtral |
+| **Azure** | `Azure` | Enterprise-grade OpenAI |
+| **OpenRouter**| `OpenRouter`| 100+ community models |
+
+---
+
+## 🗺️ Roadmap
+
+- [x] Baseline multiple providers
+- [x] Automated Retry & Timeout controls
+- [ ] Async API (`await ask_async`)
+- [ ] Provider Fallback (`fallback=[Groq()]`)
+- [ ] Structured Outputs (Pydantic Support)
+
+---
+
+## 🔗 Important Links
+
+- **GitHub Repository**: [Hosseinghorbani0/ask-ai](https://github.com/Hosseinghorbani0/ask-ai) (Star us! ⭐)
+- **Official Website**: [hosseinghorbani0.ir](https://hosseinghorbani0.ir/)
+- **Bug Tracker**: [Report an Issue](https://github.com/Hosseinghorbani0/ask-ai/issues)

{askai_python-0.1.6.dist-info → askai_python-0.2.1.dist-info}/RECORD

@@ -1,10 +1,11 @@
 ask_ai/__init__.py,sha256=i08NdBtFsjV3vT9ZqhEsroXd7vvZrWDojsOW-79iTV4,439
 ask_ai/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-ask_ai/base.py,sha256=2-SnoQiy-RAmjCceiAsx-rqwbh20xAMpLdtPiiJdNUk,5389
+ask_ai/base.py,sha256=pMNc6YhHiWasi1IH6Qe6xdtNwoxzQu68JPW5KBrZCsE,8543
 ask_ai/cli.py,sha256=8oeRLsT2lrcuG0vDq_a4CVqvzzmwDG3HxHW4BYg3IxM,2708
 ask_ai/config.py,sha256=oP5RjdbbdDve8G-FlcbNaMXPorBw96GxhCbfFWnfbIw,1934
-ask_ai/exceptions.py,sha256=JT2XbhsaWCFn3bPJD37RVH9HehV1ngus5s05v6xc1Tg,414
+ask_ai/exceptions.py,sha256=jUP1tJJvvx8ymFXU19Em3JBbTPixEnO_uj6ZC5zDyEM,893
 ask_ai/media.py,sha256=cxPZ0XoY0iU3Y8bY-YV3MGPlNFOJ_FjjwMATgo-jcFw,1821
+ask_ai/utils.py,sha256=cRokmKtoNOgQsGrKTdKBjv9vdG3sXpiA6W9HtcjNIho,1582
 ask_ai/providers/__init__.py,sha256=tVSW-FfSMPzCYK_TZy93W5ckdVe2PBPyzogpKXH_jhg,341
 ask_ai/providers/anthropic_provider.py,sha256=NL_6E7eNTYAoYpym4ewD8Vj8tF_ewuFjFkRR0Jj0L-U,3670
 ask_ai/providers/azure_provider.py,sha256=bY8MlHIjiH_H00Q78_2ooCgN1OPeM3ajpvJ2AkxNsTI,2086
@@ -12,8 +13,8 @@ ask_ai/providers/google_provider.py,sha256=DsIgu_cBC7ta0lKclAlge1PYfcp3zXKPLjWSe
 ask_ai/providers/groq_provider.py,sha256=iU7BTpvmBniE78VdTQETrCXrJzdXTpWH2oN9_sUgnFk,3268
 ask_ai/providers/openai_provider.py,sha256=b6grPbsUIY19HkABDrUIAZAIcJCKQup5tooU2LFlkvo,4793
 ask_ai/providers/openrouter_provider.py,sha256=LCGK-vNxy_maY2Fle6xh_i8MVm7a_bhjCaF7Uj0H3AM,1986
-askai_python-0.1.6.dist-info/licenses/LICENSE,sha256=SUSbAUsh_Uqw7NDEKwAGRVPS7sojihBHIBGBx6WZxvg,1073
-askai_python-0.1.6.dist-info/METADATA,sha256=auX_qAPfA8IY8g38KL-IQDxBddvRdIFiVDzGkzQHd3o,3850
-askai_python-0.1.6.dist-info/WHEEL,sha256=YLJXdYXQ2FQ0Uqn2J-6iEIC-3iOey8lH3xCtvFLkd8Q,91
-askai_python-0.1.6.dist-info/top_level.txt,sha256=fW2GapXaz6f-2UAuzVVPUPNfZUK_ZS1dpLyCllcE03s,7
-askai_python-0.1.6.dist-info/RECORD,,
+askai_python-0.2.1.dist-info/licenses/LICENSE,sha256=SUSbAUsh_Uqw7NDEKwAGRVPS7sojihBHIBGBx6WZxvg,1073
+askai_python-0.2.1.dist-info/METADATA,sha256=sDAgOTv2w_wTOBcJrOp5-c8io7eMgHi13kjcv2h_myE,4218
+askai_python-0.2.1.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+askai_python-0.2.1.dist-info/top_level.txt,sha256=fW2GapXaz6f-2UAuzVVPUPNfZUK_ZS1dpLyCllcE03s,7
+askai_python-0.2.1.dist-info/RECORD,,

{askai_python-0.1.6.dist-info → askai_python-0.2.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (81.0.0)
+Generator: setuptools (82.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

askai_python-0.1.6.dist-info/METADATA

@@ -1,135 +0,0 @@
-Metadata-Version: 2.4
-Name: askai-python
-Version: 0.1.6
-Summary: AI Made Stupid Simple. Unified API for OpenAI, Groq, Google, and more.
-Author-email: Hossein Ghorbani <hosseingh1068@gmail.com>
-License: MIT
-Project-URL: Homepage, https://hosseinghorbani0.ir/
-Project-URL: Bug Tracker, https://github.com/Hosseinghorbani0/ask-ai/issues
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: openai>=1.0.0
-Requires-Dist: groq
-Requires-Dist: google-generativeai
-Requires-Dist: anthropic
-Requires-Dist: requests
-Requires-Dist: Pillow
-Dynamic: license-file
-
-# askai-python 🚀
-
-**AI Made Stupid Simple.** One unified Python client for OpenAI, Claude, Gemini, Groq, and Azure.
-
-[![PyPI version](https://img.shields.io/pypi/v/askai-python.svg)](https://pypi.org/project/askai-python/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
-[![GitHub](https://img.shields.io/badge/GitHub-View%20Source-blue.svg)](https://github.com/Hosseinghorbani0/ask-ai)
-
----
-
-## ⚡ What is askai-python?
-
-`askai-python` is a lightweight, zero-dependency-logic wrapper that provides a **single API** for all major AI providers. Stop learning multiple SDKs; just call `.ask()` and get your answer.
-
-- **Unified Interface**: One syntax for OpenAI, Anthropic, Google, and more.
-- **Zero-Config**: Seamlessly uses environment variables for API keys.
-- **Smart Media**: Easy handling of images and audio (Provider permitting).
-- **Production Grade**: Built-in robust error handling and type hints.
-
----
-
-## 📦 Installation
-
-```bash
-pip install askai-python
-```
-
----
-
-## 🚀 Examples & Tutorials
-
-### 1. Basic Usage (The 3-Line Magic)
-```python
-from ask_ai import OpenAI
-
-ai = OpenAI() # Auto-loads OPENAI_API_KEY from env
-print(ai.ask("What is the capital of France?").text)
-```
-
-### 2. Multi-Provider Mastery
-Switching between top-tier models has never been easier:
-
-```python
-from ask_ai import OpenAI, Anthropic, Google
-
-prompt = "Explain quantum computing to a 5-year-old."
-
-# OpenAI (GPT-4o)
-print(OpenAI().ask(prompt).text)
-
-# Anthropic (Claude 3.5 Sonnet)
-print(Anthropic().ask(prompt).text)
-
-# Google (Gemini 1.5 Pro)
-print(Google().ask(prompt).text)
-```
-
-### 3. Advanced Configuration (System Personas)
-You can set a system prompt and temperature globally for an instance or per-request.
-
-```python
-from ask_ai import OpenAI
-
-ai = OpenAI()
-
-# Global config
-ai.advanced(
-    prompt="You are a sarcastic robot from the year 3000.",
-    temperature=0.9
-)
-
-print(ai.ask("What is love?").text)
-
-# Per-request override
-print(ai.ask("What is 1+1?", temperature=0.1).text)
-```
-
-### 4. Handling Structured Exceptions
-Build reliable apps by catching specific AI errors:
-
-```python
-from ask_ai import OpenAI
-from ask_ai.exceptions import AskAIError, APIKeyError
-
-try:
-    ai = OpenAI(api_key="invalid-key")
-    ai.ask("Hello")
-except APIKeyError:
-    print("Check your API keys!")
-except AskAIError as e:
-    print(f"An AI error occurred: {e}")
-```
-
----
-
-## 🔌 Supported Providers
-
-| Provider | Class | Capabilities |
-|----------|-------|-------------|
-| **OpenAI** | `OpenAI` | Text, Images (DALL-E), Vision |
-| **Anthropic** | `Anthropic` | Text, Vision (Claude 3.5) |
-| **Google** | `Google` | Text, Images, Video, Audio |
-| **Groq** | `Groq` | Ultra-fast Llama 3 & Mixtral |
-| **Azure** | `Azure` | Enterprise-grade OpenAI |
-| **OpenRouter**| `OpenRouter`| 100+ community models |
-
----
-
-## 🔗 Important Links
-
-- **GitHub Repository**: [Hosseinghorbani0/ask-ai](https://github.com/Hosseinghorbani0/ask-ai) (Star us! ⭐)
-- **Official Website**: [hosseinghorbani0.ir](https://hosseinghorbani0.ir/)
-- **Bug Tracker**: [Report an Issue](https://github.com/Hosseinghorbani0/ask-ai/issues)
-
----
-*Built with ❤️ by [Hossein Ghorbani](https://github.com/Hosseinghorbani0).*