Merge pull request #2303 from Agenta-AI/fix/cache-and-tree

[Fix] Remove Auth Cache + Fix t`ree=None`
Agenta-AI · Nov 25, 2024 · 35dd4fc · 35dd4fc
2 parents fdc03ac + b8e17c3
commit 35dd4fc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/agenta-cli/agenta/sdk/decorators/routing.py b/agenta-cli/agenta/sdk/decorators/routing.py
@@ -373,19 +373,19 @@ async def execute_function(
 
     async def handle_success(self, result: Any, inline_trace: bool):
         data = None
-        trace = dict()
+        tree = None
 
         with suppress():
             data = self.patch_result(result)
 
             if inline_trace:
-                trace = await self.fetch_inline_trace(inline_trace)
+                tree = await self.fetch_inline_trace(inline_trace)
 
         log.info(f"----------------------------------")
         log.info(f"Agenta SDK - exiting with success: 200")
         log.info(f"----------------------------------")
 
-        return BaseResponse(data=data, tree=trace)
+        return BaseResponse(data=data, tree=tree)
 
     def handle_failure(self, error: Exception):
         log.warning("--------------------------------------------------")

diff --git a/agenta-cli/agenta/sdk/middleware/auth.py b/agenta-cli/agenta/sdk/middleware/auth.py
@@ -27,6 +27,8 @@
     "t",
 )
 
+AGENTA_SDK_AUTH_CACHE = False
+
 AGENTA_UNAUTHORIZED_EXECUTION_ALLOWED = str(
     environ.get("AGENTA_UNAUTHORIZED_EXECUTION_ALLOWED", False)
 ).lower() in ("true", "1", "t")