10xscale-agentflow-cli 0.1.9__tar.gz → 0.2.0__tar.gz

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/10xscale_agentflow_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.1.9
+Version: 0.2.0
 Summary: CLI and API for 10xscale AgentFlow
 Author-email: 10xscale <contact@10xscale.ai>
 Maintainer-email: Shudipto Trafder <shudiptotrafder@gmail.com>

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.1.9
+Version: 0.2.0
 Summary: CLI and API for 10xscale AgentFlow
 Author-email: 10xscale <contact@10xscale.ai>
 Maintainer-email: Shudipto Trafder <shudiptotrafder@gmail.com>

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/agentflow_cli/src/app/core/auth/jwt_auth.py

@@ -44,6 +44,11 @@ class JwtAuth(BaseAuth):
         jwt_secret_key = os.environ.get("JWT_SECRET_KEY", None)
         jwt_algorithm = os.environ.get("JWT_ALGORITHM", None)
 
+        # check bearer token then remove barer prefix
+        token = credential.credentials
+        if token.lower().startswith("bearer "):
+            token = token[7:]
+
         if jwt_secret_key is None or jwt_algorithm is None:
             raise UserAccountError(
                 message="JWT settings are not configured",
@@ -52,7 +57,7 @@ class JwtAuth(BaseAuth):
 
         try:
             decoded_token = jwt.decode(
-                credential.credentials,
+                token,
                 jwt_secret_key,  # type: ignore
                 algorithms=[jwt_algorithm],  # type: ignore
             )

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/agentflow_cli/src/app/loader.py

@@ -202,6 +202,14 @@ async def attach_all_modules(
                 path,
             )
             container.bind_instance(BaseAuth, auth_backend)
+        elif method == "jwt":
+            from agentflow_cli.src.app.core.auth.jwt_auth import JwtAuth
+
+            jwt_auth = JwtAuth()
+            container.bind_instance(BaseAuth, jwt_auth)
+
+        elif method == "none":
+            container.bind_instance(BaseAuth, None, allow_none=True)
     else:
         # bind None
         container.bind_instance(BaseAuth, None, allow_none=True)

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/agentflow_cli/src/app/routers/checkpointer/services/checkpointer_service.py

@@ -174,7 +174,8 @@ class CheckpointerService:
         base: dict[str, Any] = {}
         if old_state is not None:
             # Keep full dump so we can preserve existing fields
-            base = old_state.model_dump()
+            # Use serialize_as_any=True to include subclass fields
+            base = old_state.model_dump(serialize_as_any=True)
 
         merged: dict[str, Any] = {**base}
 

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/agentflow_cli/src/app/routers/graph/services/graph_service.py

@@ -247,7 +247,7 @@ class GraphService:
 
             return GraphInvokeOutputSchema(
                 messages=messages,
-                state=raw_state.model_dump() if raw_state else None,
+                state=raw_state.model_dump(serialize_as_any=True) if raw_state else None,
                 context=context,
                 summary=context_summary,
                 meta=meta,
@@ -295,7 +295,7 @@ class GraphService:
                 mt = chunk.metadata or {}
                 mt.update(meta)
                 chunk.metadata = mt
-                yield chunk.model_dump_json()
+                yield chunk.model_dump_json(serialize_as_any=True)
                 if (
                     self.config.thread_name_generator_path
                     and meta["is_new_thread"]
@@ -317,7 +317,7 @@ class GraphService:
                     event=StreamEvent.UPDATES,
                     data={"status": "completed"},
                     metadata=meta,
-                ).model_dump_json()
+                ).model_dump_json(serialize_as_any=True)
 
         except Exception as e:
             logger.error(f"Graph streaming failed: {e}")
@@ -416,7 +416,7 @@ class GraphService:
                     "success": True,
                     "message": "No messages found in state",
                     "removed_count": 0,
-                    "state": state.model_dump_json(),
+                    "state": state.model_dump_json(serialize_as_any=True),
                 }
 
             filtered = [m for m in messages if not self._has_empty_tool_call(m)]
@@ -433,7 +433,7 @@ class GraphService:
                 "success": True,
                 "message": message,
                 "removed_count": removed_count,
-                "state": state.model_dump_json(),
+                "state": state.model_dump_json(serialize_as_any=True),
             }
         except Exception as e:
             logger.error(f"Fix graph operation failed: {e}")

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/agentflow_cli/src/app/utils/parse_output.py

@@ -7,11 +7,11 @@ from agentflow_cli.src.app.core.config.settings import Settings
 
 def parse_state_output(settings: Settings, response: BaseModel) -> dict[str, Any]:
     # if settings.IS_DEBUG:
-    #     return response.model_dump(exclude={"execution_meta"})
-    return response.model_dump()
+    #     return response.model_dump(exclude={"execution_meta"}, serialize_as_any=True)
+    return response.model_dump(serialize_as_any=True)
 
 
 def parse_message_output(settings: Settings, response: BaseModel) -> dict[str, Any]:
     # if settings.IS_DEBUG:
-    #     return response.model_dump(exclude={"raw"})
-    return response.model_dump()
+    #     return response.model_dump(exclude={"raw"}, serialize_as_any=True)
+    return response.model_dump(serialize_as_any=True)

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "10xscale-agentflow-cli"
-version = "0.1.9"
+version = "0.2.0"
 description = "CLI and API for 10xscale AgentFlow"
 readme = "README.md"
 license = {text = "MIT"}

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.0}/tests/test_utils_parse_and_callable.py

@@ -28,9 +28,7 @@ def test_parse_state_output(is_debug: bool):
     settings = Settings(IS_DEBUG=is_debug)
     model = _StateModel(a=1, b="x", execution_meta={"duration": 123})
     out = parse_state_output(settings, model)
-    # execution_meta excluded only in debug mode per implementation
-    # Since parse_state_output doesn't filter execution_meta (commented out),
-    # it should always be present regardless of debug mode
+    # Current implementation always includes execution_meta regardless of debug
     assert out["execution_meta"] == {"duration": 123}
     assert out["a"] == 1 and out["b"] == "x"
 
@@ -40,8 +38,7 @@ def test_parse_message_output(is_debug: bool):
     settings = Settings(IS_DEBUG=is_debug)
     model = _MessageModel(content="hello", raw={"tokens": 5})
     out = parse_message_output(settings, model)
-    # Since parse_message_output doesn't filter raw (commented out),
-    # it should always be present regardless of debug mode
+    # Current implementation always includes raw regardless of debug
     assert out["raw"] == {"tokens": 5}
     assert out["content"] == "hello"