trustgraph-flow 2.7.1__tar.gz → 2.7.2__tar.gz

{trustgraph_flow-2.7.1 → trustgraph_flow-2.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-flow
-Version: 2.7.1
+Version: 2.7.2
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_flow-2.7.2/trustgraph/flow_version.py

@@ -0,0 +1 @@
+__version__ = "2.7.2"

{trustgraph_flow-2.7.1 → trustgraph_flow-2.7.2}/trustgraph/gateway/auth.py

@@ -189,36 +189,44 @@ class IamAuth:
     # Lifecycle
     # ------------------------------------------------------------------
 
+    async def _fetch_signing_key(self):
+        """Single attempt to fetch the IAM signing public key.
+        Returns True on success."""
+        try:
+            async def _fetch(client):
+                return await client.get_signing_key_public()
+
+            pem = await self._with_client(_fetch)
+            if pem:
+                self._signing_public_pem = pem
+                logger.info(
+                    "IamAuth: fetched IAM signing public key "
+                    f"({len(pem)} bytes)"
+                )
+                return True
+        except Exception as e:
+            logger.debug(
+                f"IamAuth: signing key fetch failed: "
+                f"{type(e).__name__}: {e}"
+            )
+        return False
+
     async def start(self, max_retries=30, retry_delay=2.0):
         """Fetch the signing public key from IAM.  Retries on
         failure — the gateway may be starting before IAM is ready."""
 
-        async def _fetch(client):
-            return await client.get_signing_key_public()
-
         for attempt in range(max_retries):
-            try:
-                pem = await self._with_client(_fetch)
-                if pem:
-                    self._signing_public_pem = pem
-                    logger.info(
-                        "IamAuth: fetched IAM signing public key "
-                        f"({len(pem)} bytes)"
-                    )
-                    return
-            except Exception as e:
-                logger.info(
-                    f"IamAuth: waiting for IAM signing key "
-                    f"({type(e).__name__}: {e}); "
-                    f"retry {attempt + 1}/{max_retries}"
-                )
+            if await self._fetch_signing_key():
+                return
+            logger.info(
+                f"IamAuth: waiting for IAM signing key; "
+                f"retry {attempt + 1}/{max_retries}"
+            )
             await asyncio.sleep(retry_delay)
 
-        # Don't prevent startup forever.  A later authenticate() call
-        # will try again via the JWT path.
         logger.warning(
             "IamAuth: could not fetch IAM signing key at startup; "
-            "JWT validation will fail until it's available"
+            "JWT validation will retry on first request"
         )
 
     # ------------------------------------------------------------------
@@ -259,7 +267,7 @@ class IamAuth:
             self._annotate_request(request, identity)
             return identity
         if token.count(".") == 2:
-            identity = self._verify_jwt(token)
+            identity = await self._verify_jwt(token)
             self._annotate_request(request, identity)
             return identity
         raise _auth_failure()
@@ -281,7 +289,9 @@ class IamAuth:
         except Exception:
             pass
 
-    def _verify_jwt(self, token):
+    async def _verify_jwt(self, token):
+        if not self._signing_public_pem:
+            await self._fetch_signing_key()
         if not self._signing_public_pem:
             raise _auth_failure()
         try:

{trustgraph_flow-2.7.1 → trustgraph_flow-2.7.2}/trustgraph/retrieval/document_rag/rag.py

@@ -36,6 +36,7 @@ class Processor(FlowProcessor):
         fetch_limit = params.get("fetch_limit", 0)
         rerank_diversity_mode = params.get("rerank_diversity_mode", "none")
         rerank_diversity_lambda = params.get("rerank_diversity_lambda", 0.7)
+        fetch_chunk_timeout = params.get("fetch_chunk_timeout", 120)
         retrieval_mode = params.get("retrieval_mode", "vector")
         vector_weight = params.get("vector_weight", 1.0)
         keyword_weight = params.get("keyword_weight", 1.0)
@@ -47,6 +48,7 @@ class Processor(FlowProcessor):
                 "fetch_limit": fetch_limit,
                 "rerank_diversity_mode": rerank_diversity_mode,
                 "rerank_diversity_lambda": rerank_diversity_lambda,
+                "fetch_chunk_timeout": fetch_chunk_timeout,
                 "retrieval_mode": retrieval_mode,
                 "vector_weight": vector_weight,
                 "keyword_weight": keyword_weight,
@@ -57,6 +59,7 @@ class Processor(FlowProcessor):
         self.fetch_limit = fetch_limit
         self.rerank_diversity_mode = rerank_diversity_mode
         self.rerank_diversity_lambda = rerank_diversity_lambda
+        self.fetch_chunk_timeout = fetch_chunk_timeout
         self.retrieval_mode = retrieval_mode
         self.vector_weight = vector_weight
         self.keyword_weight = keyword_weight
@@ -139,7 +142,7 @@ class Processor(FlowProcessor):
 
             logger.info(f"Handling input {id}...")
 
-            async def fetch_chunk(chunk_id, timeout=120):
+            async def fetch_chunk(chunk_id, timeout=self.fetch_chunk_timeout):
                 return await flow.librarian.fetch_document_text(
                     document_id=chunk_id, timeout=timeout,
                 )
@@ -329,6 +332,14 @@ class Processor(FlowProcessor):
             help='MMR relevance/diversity tradeoff, higher values prefer relevance'
         )
 
+        parser.add_argument(
+            '--fetch-chunk-timeout',
+            type=int,
+            default=120,
+            help='Timeout in seconds for fetching a document chunk from the '
+                 'librarian (default: 120)'
+        )
+        
         parser.add_argument(
             '--retrieval-mode',
             choices=['vector', 'keyword', 'hybrid'],

{trustgraph_flow-2.7.1 → trustgraph_flow-2.7.2}/trustgraph_flow.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trustgraph-flow
-Version: 2.7.1
+Version: 2.7.2
 Summary: TrustGraph provides a means to run a pipeline of flexible AI processing components in a flexible means to achieve a processing pipeline.
 Author-email: "trustgraph.ai" <security@trustgraph.ai>
 Project-URL: Homepage, https://github.com/trustgraph-ai/trustgraph

trustgraph_flow-2.7.1/trustgraph/flow_version.py

@@ -1 +0,0 @@
-__version__ = "2.7.1"