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

{10xscale_agentflow_cli-0.2.1 → 10xscale_agentflow_cli-0.2.3}/10xscale_agentflow_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.2.1
+Version: 0.2.3
 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.1 → 10xscale_agentflow_cli-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.2.1
+Version: 0.2.3
 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.1 → 10xscale_agentflow_cli-0.2.3}/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.1 → 10xscale_agentflow_cli-0.2.3}/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.1 → 10xscale_agentflow_cli-0.2.3}/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.1 → 10xscale_agentflow_cli-0.2.3}/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.1 → 10xscale_agentflow_cli-0.2.3}/agentflow_cli/src/app/routers/graph/services/graph_service.py

@@ -217,6 +217,7 @@ class GraphService:
             input_data, config, meta = await self._prepare_input(graph_input)
             # add user inside config
             config["user"] = user
+            config["user_id"] = user.get("id", "anonymous")
 
             # if its a new thread then save the thread into db
             await self._save_thread(config, config["thread_id"])
@@ -281,6 +282,8 @@ class GraphService:
             input_data, config, meta = await self._prepare_input(graph_input)
             # add user inside config
             config["user"] = user
+            config["user_id"] = user.get("id", "anonymous")
+
             await self._save_thread(config, config["thread_id"])
 
             messages_str = []
@@ -397,6 +400,7 @@ class GraphService:
             logger.debug(f"User info: {user}")
 
             fix_config = {"thread_id": thread_id, "user": user}
+            fix_config["user_id"] = user.get("id", "anonymous")
             if config:
                 fix_config.update(config)
 

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

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