agstack 1.2.2__tar.gz → 1.2.3__tar.gz

{agstack-1.2.2 → agstack-1.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.2.2
+Version: 1.2.3
 Summary: Production-ready toolkit for building FastAPI and LLM applications
 Author-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>
 Maintainer-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>

{agstack-1.2.2 → agstack-1.2.3}/agstack/llm/client.py

@@ -4,6 +4,7 @@ import logging
 import time
 from typing import TYPE_CHECKING, Any, AsyncIterator, Literal, overload
 
+import httpx
 from httpx import AsyncClient
 from httpx import Timeout as HttpxTimeout
 from openai import APIError, APITimeoutError, AsyncOpenAI, OpenAI, RateLimitError
@@ -447,7 +448,15 @@ class LLMClient:
         :param model: 模型名称
         :return: [(index, score, text), ...] 按相关性降序排列
         """
+        from httpx import ConnectTimeout, TimeoutException
 
+        @autoretry(
+            logger,
+            retries=3,
+            delay=2.0,
+            backoff=2.0,
+            exceptions=(ConnectTimeout, ConnectionError),
+        )
         async def _call():
             return await self._async_http_client.post(
                 f"{self._base_url}/reranks",
@@ -463,6 +472,7 @@ class LLMClient:
 
         try:
             response = await _call()
+            response.raise_for_status()
             data = response.json()
 
             # 解析响应
@@ -476,6 +486,17 @@ class LLMClient:
 
             return results
 
+        except TimeoutException as e:
+            logger.error(f"Rerank timeout: {e}")
+            raise LLMTimeoutError("LLM_RERANK_TIMEOUT") from e
+
+        except httpx.HTTPStatusError as e:
+            if e.response.status_code == 429:
+                logger.error(f"Rerank rate limit: {e}")
+                raise LLMRateLimitError("LLM_RERANK_RATE_LIMIT") from e
+            logger.error(f"Rerank HTTP error {e.response.status_code}: {e}")
+            raise LLMError("LLM_RERANK_ERROR", {"error": str(e)}, http_status=e.response.status_code) from e
+
         except Exception as e:
             logger.error(f"Rerank error: {e}")
             raise LLMError("LLM_RERANK_ERROR", {"error": str(e)}) from e
@@ -495,6 +516,8 @@ class LLMClient:
         :param model: 模型名称
         :return: [(index, score, text), ...] 按相关性降序排列
         """
+        import requests
+
         session = self._get_sync_http_session()
 
         try:
@@ -522,6 +545,18 @@ class LLMClient:
 
             return results
 
+        except requests.Timeout as e:
+            logger.error(f"Rerank timeout (sync): {e}")
+            raise LLMTimeoutError("LLM_RERANK_TIMEOUT") from e
+
+        except requests.HTTPError as e:
+            status = e.response.status_code if e.response is not None else 500
+            if status == 429:
+                logger.error(f"Rerank rate limit (sync): {e}")
+                raise LLMRateLimitError("LLM_RERANK_RATE_LIMIT") from e
+            logger.error(f"Rerank HTTP error {status} (sync): {e}")
+            raise LLMError("LLM_RERANK_ERROR", {"error": str(e)}, http_status=status) from e
+
         except Exception as e:
             logger.error(f"Rerank error (sync): {e}")
             raise LLMError("LLM_RERANK_ERROR", {"error": str(e)}) from e

{agstack-1.2.2 → agstack-1.2.3}/agstack/security/casbin.py

@@ -40,7 +40,7 @@ class CasbinRules(ObjectModel):
     created_at: Column[datetime] = column(type="datetime", default_factory=datetime.now)
 
     class Config:
-        table_name = "system_casbin_rule"
+        table_name = "system_casbin_rules"
 
 
 class SqlObjectsAdapter(AsyncAdapter, AsyncFilteredAdapter):

{agstack-1.2.2 → agstack-1.2.3}/agstack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agstack
-Version: 1.2.2
+Version: 1.2.3
 Summary: Production-ready toolkit for building FastAPI and LLM applications
 Author-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>
 Maintainer-email: XtraVisions <gitadmin@xtravisions.com>, Chen Hao <chenhao@xtravisions.com>

{agstack-1.2.2 → agstack-1.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agstack"
-version = "1.2.2"
+version = "1.2.3"
 description = "Production-ready toolkit for building FastAPI and LLM applications"
 readme = "README.md"
 license = "MIT"