davidkhala.ai 0.0.4__tar.gz → 0.0.6__tar.gz

{davidkhala_ai-0.0.4 → davidkhala_ai-0.0.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: davidkhala.ai
-Version: 0.0.4
+Version: 0.0.6
 Summary: misc AI modules
 Requires-Python: >=3.13
 Provides-Extra: ali

{davidkhala_ai-0.0.4 → davidkhala_ai-0.0.6}/davidkhala/ai/api/__init__.py

@@ -38,7 +38,8 @@ class API(AbstractClient):
             "meta": {
                 "usage": response['usage'],
                 "created": datetime.datetime.fromtimestamp(response['created'])
-            }
+            },
+            'model': response['model'],
         }
 
     def list_models(self):

{davidkhala_ai-0.0.4 → davidkhala_ai-0.0.6}/davidkhala/ai/api/openrouter.py

@@ -1,9 +1,12 @@
+import time
+
 import requests
 from davidkhala.http_request import default_on_response
-from requests import HTTPError, Response
+from requests import Response
 
 from davidkhala.ai.api import API
 
+
 class OpenRouter(API):
     @property
     def free_models(self) -> list[str]:
@@ -20,11 +23,11 @@ class OpenRouter(API):
         err = r.get('error')
         if err:
             derived_response = Response()
-            derived_response.status_code = err.pop('code', None)
-            derived_response._content = err.pop('message', None)
-            http_err =  HTTPError(response=derived_response)
-            http_err.metadata = err.get("metadata")
-            raise http_err
+            derived_response.status_code = err['code']
+            derived_response.reason = err['message']
+            derived_response.metadata = err.get("metadata")
+
+            derived_response.raise_for_status()
         return r
 
     def __init__(self, api_key: str, *models: str, **kwargs):
@@ -36,16 +39,23 @@ class OpenRouter(API):
                 'url']  # Site URL for rankings on openrouter.ai.
             self._.options["headers"]["X-Title"] = kwargs['leaderboard'][
                 'name']  # Site title for rankings on openrouter.ai.
-        if not models:
-            models = [self.free_models[0]]
         self.models = models
 
         self._.on_response = OpenRouter.on_response
 
     def chat(self, *user_prompt: str, **kwargs):
-        if len(self.models) > 1:
+        if self.models:
             kwargs["models"] = self.models
         else:
-            kwargs["model"] = self.models[0]
-
-        return super().chat(*user_prompt, **kwargs)
+            kwargs["model"] = self.model
+
+        try:
+            r = super().chat(*user_prompt, **kwargs)
+        except requests.HTTPError as e:
+            if e.response.status_code == 429 and kwargs.get('retry'):
+                time.sleep(1)
+                return self.chat(*user_prompt, **kwargs)
+            else: raise
+        if self.models:
+            assert r['model'] in self.models
+        return r

{davidkhala_ai-0.0.4 → davidkhala_ai-0.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "davidkhala.ai"
-version = "0.0.4"
+version = "0.0.6"
 description = "misc AI modules"
 readme = "README.md"
 requires-python = ">=3.13"