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

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

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

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

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

{10xscale_agentflow_cli-0.2.0 → 10xscale_agentflow_cli-0.2.2}/agentflow_cli/src/app/core/auth/auth_backend.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from fastapi import Depends, Response
+from fastapi import Depends, Request, Response
 from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
 from injectq.integrations import InjectAPI
 
@@ -10,7 +10,8 @@ from agentflow_cli.src.app.core.config.graph_config import GraphConfig
 
 
 def verify_current_user(
-    res: Response,
+    request: Request,
+    response: Response,
     credential: HTTPAuthorizationCredentials = Depends(
         HTTPBearer(auto_error=False),
     ),
@@ -27,7 +28,11 @@ def verify_current_user(
         logger.error("Auth backend is not configured")
         return user
 
-    user: dict | None = auth_backend.authenticate(res, credential)
+    user: dict | None = auth_backend.authenticate(
+        request,
+        response,
+        credential,
+    )
     if user and "user_id" not in user:
         logger.error("Authentication failed: 'user_id' not found in user info")
     return user or {}

{10xscale_agentflow_cli-0.2.0 → 10xscale_agentflow_cli-0.2.2}/agentflow_cli/src/app/core/auth/base_auth.py

@@ -1,14 +1,17 @@
 from abc import ABC, abstractmethod
 from typing import Any
 
-from fastapi import Response
+from fastapi import Request, Response
 from fastapi.security import HTTPAuthorizationCredentials
 
 
 class BaseAuth(ABC):
     @abstractmethod
     def authenticate(
-        self, res: Response, credential: HTTPAuthorizationCredentials
+        self,
+        request: Request,
+        response: Response,
+        credential: HTTPAuthorizationCredentials,
     ) -> dict[str, Any] | None:
         """Authenticate the user based on the provided credentials.
         IT should return an empty dict if no authentication is required.

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

@@ -2,7 +2,7 @@ import os
 from typing import Any
 
 import jwt
-from fastapi import Response
+from fastapi import Request, Response
 from fastapi.security import HTTPAuthorizationCredentials
 
 from agentflow_cli.src.app.core import logger
@@ -12,7 +12,10 @@ from agentflow_cli.src.app.core.exceptions import UserAccountError
 
 class JwtAuth(BaseAuth):
     def authenticate(
-        self, res: Response, credential: HTTPAuthorizationCredentials
+        self,
+        request: Request,
+        response: Response,
+        credential: HTTPAuthorizationCredentials,
     ) -> dict[str, Any] | None:
         """No authentication is required, so return None."""
         """
@@ -72,7 +75,8 @@ class JwtAuth(BaseAuth):
                 message="Invalid token, please login again",
                 error_code="INVALID_TOKEN",
             )
-        res.headers["WWW-Authenticate"] = 'Bearer realm="auth_required"'
+
+        response.headers["WWW-Authenticate"] = 'Bearer realm="auth_required"'
 
         # check if user_id exists in the token
         if "user_id" not in decoded_token:

{10xscale_agentflow_cli-0.2.0 → 10xscale_agentflow_cli-0.2.2}/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.2.0 → 10xscale_agentflow_cli-0.2.2}/agentflow_cli/src/app/loader.py

@@ -1,6 +1,8 @@
 import importlib
 import inspect
 import logging
+import os
+from pathlib import Path
 
 from agentflow.checkpointer import BaseCheckpointer
 from agentflow.graph import CompiledGraph
@@ -197,6 +199,24 @@ async def attach_all_modules(
     if auth_config:
         method = auth_config.get("method", None)
         path = auth_config.get("path", None)
+        if not path or not method:
+            raise ValueError("Both 'method' and 'path' must be specified in auth_config.")
+
+        # Extract file path before the ':' for existence check
+        module_or_path = path.split(":", 1)[0] if ":" in path else path
+
+        # Simple handling: if it appears to be a filesystem path, use it; otherwise
+        # convert dotted module path to a file path like src/auth/custom_auth.py
+        if os.path.sep in module_or_path or module_or_path.endswith(".py"):
+            file_path = Path(module_or_path)
+        elif "." in module_or_path and os.path.sep not in module_or_path:
+            file_path = Path(module_or_path.replace(".", os.path.sep) + ".py")
+        else:
+            file_path = Path(module_or_path)
+
+        if not file_path.exists():
+            raise ValueError(f"Custom auth path does not exist: {module_or_path}")
+
         if method == "custom":
             auth_backend = load_auth(
                 path,

{10xscale_agentflow_cli-0.2.0 → 10xscale_agentflow_cli-0.2.2}/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.2.0 → 10xscale_agentflow_cli-0.2.2}/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
@@ -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(serialize_as_any=True)
+                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(serialize_as_any=True)
+                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}")

{10xscale_agentflow_cli-0.2.0 → 10xscale_agentflow_cli-0.2.2}/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.2.0 → 10xscale_agentflow_cli-0.2.2}/pyproject.toml

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