askai-python 0.2.0__py3-none-any.whl → 0.3.3__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
@@ -96,7 +117,10 @@ class BaseProvider:
         last_error = None
         while attempts <= retry:
             try:
-                return self._send_request(messages, final_config, output_type)
+                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:
@@ -112,6 +136,17 @@ class BaseProvider:
         from .exceptions import AskAINetworkError
         raise AskAINetworkError(f"Request failed after {retry} retries. Last error: {last_error}")
 
+    async def ask_async(self, query: str, **kwargs) -> Response:
+        """
+        Asynchronous wrapper for the .ask() method.
+        Perfect for Telegram bots (Aiogram) and FastAPI.
+        """
+        import asyncio
+        loop = asyncio.get_event_loop()
+        def _sync_ask():
+            return self.ask(query, **kwargs)
+        return await loop.run_in_executor(None, _sync_ask)
+
     def _is_retryable_error(self, e: Exception) -> bool:
         """Dynamically detect rate limits and network timeouts across different provider SDKs."""
         err_str = str(e).lower()
@@ -133,10 +168,16 @@ class BaseProvider:
 
     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

@@ -22,5 +22,9 @@ 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/requirements.txt

@@ -0,0 +1,5 @@
+openai>=1.0.0
+groq
+google-generativeai
+anthropic
+requests

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.0.dist-info → askai_python-0.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: askai-python
-Version: 0.2.0
+Version: 0.3.3
 Summary: AI Made Stupid Simple. Unified API for OpenAI, Groq, Google, and more.
 Author-email: Hossein Ghorbani <hosseingh1068@gmail.com>
 License: MIT
@@ -53,16 +53,6 @@ print(Groq().ask("Explain black holes like I'm 5").text)
 - **Zero config**: Keys are pulled from the environment automatically
 - **SDK-first, not a framework**: It stays out of your way.
 
-## ⚖️ How it compares
-
-| Feature        | ask-ai | LangChain |
-| -------------- | ------ | --------- |
-| Setup time     | 30 sec | 1 hour    |
-| Learning curve | ⭐      | ⭐⭐⭐⭐⭐     |
-| Async support  | ⏳ *(Coming soon)* | ⚠️ Complex |
-| Retry/Timeout  | ✅ Built-in | ❌ Manual |
-| Gateway needed | ❌ No      | ❌ No      |
-
 ## 🚫 What this project is NOT
 
 > ❌ Not an AI framework  
@@ -73,8 +63,43 @@ It does one thing perfectly: **Simplifying the API call to LLMs.**
 
 ---
 
+## 🛠️ Advanced Usage
+
+### 🧰 Developer QoL Utilities (Auto-Parsing)
+
+<p align="center">
+  <img src="https://raw.githubusercontent.com/Hosseinghorbani0/ask-ai/main/assets/features.svg" alt="ask-ai features parsing banner" width="100%">
+</p>
+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)
 
+<p align="center">
+  <img src="https://raw.githubusercontent.com/Hosseinghorbani0/ask-ai/main/assets/resiliency.svg" alt="ask-ai resiliency banner" width="100%">
+</p>
+
 Build reliable apps without writing your own loops. `askai-python` handles rate limits (`429`) and network drops via an internal exponential backoff.
 
 ```python

{askai_python-0.2.0.dist-info → askai_python-0.3.3.dist-info}/RECORD

@@ -1,10 +1,12 @@
 ask_ai/__init__.py,sha256=i08NdBtFsjV3vT9ZqhEsroXd7vvZrWDojsOW-79iTV4,439
 ask_ai/__main__.py,sha256=MSmt_5Xg84uHqzTN38JwgseJK8rsJn_11A8WD99VtEo,61
-ask_ai/base.py,sha256=N3U_O_alfhTyiIz_ts-z7YYdoJlR3yJcQxvD3Z9NNpY,7402
+ask_ai/base.py,sha256=Hxk7hFJqK8uBSabVBAMrLziPdEo0QWrGrGgpIlIEcN8,8934
 ask_ai/cli.py,sha256=8oeRLsT2lrcuG0vDq_a4CVqvzzmwDG3HxHW4BYg3IxM,2708
 ask_ai/config.py,sha256=oP5RjdbbdDve8G-FlcbNaMXPorBw96GxhCbfFWnfbIw,1934
-ask_ai/exceptions.py,sha256=V7IVUL-oNkUYXnek0J4Bt66MaxgvJQCVk8bBTpqsMk4,769
+ask_ai/exceptions.py,sha256=jUP1tJJvvx8ymFXU19Em3JBbTPixEnO_uj6ZC5zDyEM,893
 ask_ai/media.py,sha256=cxPZ0XoY0iU3Y8bY-YV3MGPlNFOJ_FjjwMATgo-jcFw,1821
+ask_ai/requirements.txt,sha256=Nvs7a0Ol8J6bTUWUJ02VqjNo9jazd6houEmDLqu-Qcw,58
+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 +14,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.2.0.dist-info/licenses/LICENSE,sha256=SUSbAUsh_Uqw7NDEKwAGRVPS7sojihBHIBGBx6WZxvg,1073
-askai_python-0.2.0.dist-info/METADATA,sha256=EHGGvSOedlST4YqUG8KD3uT0rmmK3VSJt3A35dAfjqM,3609
-askai_python-0.2.0.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
-askai_python-0.2.0.dist-info/top_level.txt,sha256=fW2GapXaz6f-2UAuzVVPUPNfZUK_ZS1dpLyCllcE03s,7
-askai_python-0.2.0.dist-info/RECORD,,
+askai_python-0.3.3.dist-info/licenses/LICENSE,sha256=SUSbAUsh_Uqw7NDEKwAGRVPS7sojihBHIBGBx6WZxvg,1073
+askai_python-0.3.3.dist-info/METADATA,sha256=o3baxmEPjdg7wflgM-aLxRL-FVuw5Q4tIJL8bVCCh90,4558
+askai_python-0.3.3.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+askai_python-0.3.3.dist-info/top_level.txt,sha256=fW2GapXaz6f-2UAuzVVPUPNfZUK_ZS1dpLyCllcE03s,7
+askai_python-0.3.3.dist-info/RECORD,,