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

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.1}/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.1
 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.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.1.9
+Version: 0.2.1
 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.1}/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.1}/agentflow_cli/src/app/core/config/setup_middleware.py

@@ -7,11 +7,38 @@ from fastapi.middleware.trustedhost import TrustedHostMiddleware
 from starlette.middleware.base import BaseHTTPMiddleware
 from starlette.middleware.cors import CORSMiddleware
 from starlette.requests import Request
+from starlette.types import ASGIApp, Receive, Scope, Send
 
 from .sentry_config import init_sentry
 from .settings import get_settings, logger
 
 
+# Paths that should be excluded from GZip compression (streaming endpoints)
+GZIP_EXCLUDED_PATHS = frozenset({"/v1/graph/stream"})
+
+
+class SelectiveGZipMiddleware:
+    """
+    GZip middleware that excludes certain paths from compression.
+
+    This is necessary because streaming endpoints need to send data
+    immediately without buffering, but GZipMiddleware buffers the
+    entire response before compressing.
+    """
+
+    def __init__(self, app: ASGIApp, minimum_size: int = 1000):
+        self.app = app
+        self.gzip_app = GZipMiddleware(app, minimum_size=minimum_size)
+
+    async def __call__(self, scope: Scope, receive: Receive, send: Send) -> None:
+        if scope["type"] == "http" and scope["path"] in GZIP_EXCLUDED_PATHS:
+            # Skip GZip for excluded paths - pass through directly
+            await self.app(scope, receive, send)
+        else:
+            # Apply GZip for all other paths
+            await self.gzip_app(scope, receive, send)
+
+
 class RequestIDMiddleware(BaseHTTPMiddleware):
     """
     Middleware to add a unique request ID and timestamp to each request and response.
@@ -74,7 +101,7 @@ def setup_middleware(app: FastAPI):
     Middleware:
         - CORS: Configured based on settings.ORIGINS.
         - TrustedHost: Configured with allowed hosts from settings.ALLOWED_HOST.
-        - GZip: Applied with a minimum size of 1000 bytes.
+        - GZip: Applied with a minimum size of 1000 bytes (excludes streaming endpoints).
     """
     settings = get_settings()
     # init cors
@@ -90,8 +117,9 @@ def setup_middleware(app: FastAPI):
 
     app.add_middleware(RequestIDMiddleware)
 
-    # Note: If you need streaming responses, you should not use GZipMiddleware.
-    app.add_middleware(GZipMiddleware, minimum_size=1000)
+    # Use SelectiveGZipMiddleware to exclude streaming endpoints from compression
+    # Streaming endpoints need immediate data transmission without buffering
+    app.add_middleware(SelectiveGZipMiddleware, minimum_size=1000)
     logger.debug("Middleware set up")
 
     # Initialize Sentry

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.1}/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.1}/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.1}/agentflow_cli/src/app/routers/graph/router.py

@@ -81,11 +81,12 @@ async def stream_graph(
 
     return StreamingResponse(
         result,
-        media_type="text/plain",
+        media_type="text/event-stream",
         headers={
-            "Cache-Control": "no-cache",
+            "Cache-Control": "no-cache, no-transform",
             "Connection": "keep-alive",
             "X-Accel-Buffering": "no",  # Disable nginx buffering
+            "Content-Encoding": "identity",  # Disable any content encoding (bypasses GZip)
         },
     )
 

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

@@ -10,7 +10,6 @@ from agentflow.utils.thread_info import ThreadInfo
 from fastapi import HTTPException
 from injectq import InjectQ, inject, singleton
 from pydantic import BaseModel
-from starlette.responses import Content
 
 from agentflow_cli.src.app.core import logger
 from agentflow_cli.src.app.core.config.graph_config import GraphConfig
@@ -247,7 +246,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,
@@ -261,7 +260,7 @@ class GraphService:
         self,
         graph_input: GraphInputSchema,
         user: dict[str, Any],
-    ) -> AsyncIterable[Content]:
+    ) -> AsyncIterable[str]:
         """
         Streams the graph execution with the provided input.
 
@@ -270,7 +269,7 @@ class GraphService:
             stream_mode (str): The stream mode ("values", "updates", "messages", etc.).
 
         Yields:
-            GraphStreamChunkSchema: Individual chunks from graph execution.
+            str: Individual JSON chunks from graph execution with newline delimiters.
 
         Raises:
             HTTPException: If graph streaming fails.
@@ -295,7 +294,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) + "\n"
                 if (
                     self.config.thread_name_generator_path
                     and meta["is_new_thread"]
@@ -313,11 +312,14 @@ class GraphService:
                 )
                 meta["thread_name"] = thread_name
 
-                yield StreamChunk(
-                    event=StreamEvent.UPDATES,
-                    data={"status": "completed"},
-                    metadata=meta,
-                ).model_dump_json()
+                yield (
+                    StreamChunk(
+                        event=StreamEvent.UPDATES,
+                        data={"status": "completed"},
+                        metadata=meta,
+                    ).model_dump_json(serialize_as_any=True)
+                    + "\n"
+                )
 
         except Exception as e:
             logger.error(f"Graph streaming failed: {e}")
@@ -416,7 +418,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 +435,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.1}/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.1}/graph/react.py

@@ -1,4 +1,5 @@
 import logging
+from time import sleep
 from typing import Any
 
 from agentflow.adapters.llm.model_response_converter import ModelResponseConverter
@@ -51,6 +52,7 @@ def get_weather(
         logger.debug("Number of messages in context: %d", len(state.context))
 
     # Mock weather response - in production, this would call a real weather API
+    sleep(1)  # Simulate network delay
     return f"The weather in {location} is sunny"
 
 

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

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

{10xscale_agentflow_cli-0.1.9 → 10xscale_agentflow_cli-0.2.1}/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"