davidkhala.ai 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

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/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.dist-info → davidkhala_ai-0.0.6.dist-info}/METADATA

@@ -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.dist-info → davidkhala_ai-0.0.6.dist-info}/RECORD

@@ -5,8 +5,8 @@ davidkhala/ai/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSu
 davidkhala/ai/agent/langgraph.py,sha256=jrc_Yvgo7eJjd3y5UJn0t1FzpnObDGYscwgsuVl2O_I,1052
 davidkhala/ai/ali/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 davidkhala/ai/ali/dashscope.py,sha256=b49unSTVXhjHCA0F1Qt6_-Zs7zjSgEetIcVK5E2nDMQ,1968
-davidkhala/ai/api/__init__.py,sha256=bcrAFJKlhLHW5CB0aqjQ04FginKVwifxVQeHF0nRRCE,1274
-davidkhala/ai/api/openrouter.py,sha256=sQ5Yef9RccKB-qCCinK4uOooyasHmlJ6f2BtO_Ly0CQ,1870
+davidkhala/ai/api/__init__.py,sha256=Ekz1jPAkDPzeNmhlp5HUSFjbeuniXLd_y5JlpGJUEcY,1316
+davidkhala/ai/api/openrouter.py,sha256=qELzeKmWLviscmBYrB9kOnEQMYAQLaJDgoqC9TB319o,2072
 davidkhala/ai/api/siliconflow.py,sha256=B5Zccpq1peNY5mQJS7duBEonj-aeQVxvmU7F7AVIN_s,1356
 davidkhala/ai/google/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 davidkhala/ai/google/adk.py,sha256=QwxYoOzT2Hol03V4NM0PF_HAzUGb4fB18VUAYacYbAY,657
@@ -14,6 +14,6 @@ davidkhala/ai/google/gemini.py,sha256=Xf4HDOOcK4-jEBERzuLnQNFsU61P2fFx4K0z-ijvNH
 davidkhala/ai/openai/__init__.py,sha256=B0LG4h4cquK2sS4DqBywV-Kf6pwOVp8hU5H8hUp-pXA,1870
 davidkhala/ai/openai/azure.py,sha256=jAtZNW1d8ub3odhB_f1MXxM7MImg64MXDnqLm1-Idkk,828
 davidkhala/ai/openai/native.py,sha256=-XuHn0KSQnuFdJ4jTwZla1ts5iOW2L2Hosfo8o8iip8,257
-davidkhala_ai-0.0.4.dist-info/METADATA,sha256=dGW0vJNd8_iDS_dXSQtvmkzCX2VqhZLFUVDWPKRgdfU,698
-davidkhala_ai-0.0.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-davidkhala_ai-0.0.4.dist-info/RECORD,,
+davidkhala_ai-0.0.6.dist-info/METADATA,sha256=M7AICLOOZqW4pG2-cxqDmFcd-dVf-nQA9grqbGh0r1o,698
+davidkhala_ai-0.0.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+davidkhala_ai-0.0.6.dist-info/RECORD,,