10xscale-agentflow-cli 0.1.6__tar.gz → 0.1.7__tar.gz

{10xscale_agentflow_cli-0.1.6 → 10xscale_agentflow_cli-0.1.7}/10xscale_agentflow_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.1.6
+Version: 0.1.7
 Summary: CLI and API for 10xscale AgentFlow
 Author-email: 10xscale <contact@10xscale.ai>
 Maintainer-email: Shudipto Trafder <shudiptotrafder@gmail.com>
@@ -26,24 +26,24 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: 10xscale-agentflow>=0.4.0
 Requires-Dist: fastapi
-Requires-Dist: gunicorn
+Requires-Dist: gunicorn==23.0.0
 Requires-Dist: orjson
-Requires-Dist: python-multipart
-Requires-Dist: pydantic
-Requires-Dist: pydantic-settings
-Requires-Dist: uvicorn
+Requires-Dist: python-multipart>=0.0.19
+Requires-Dist: pydantic>=2.8.2
+Requires-Dist: pydantic-settings>=2.3.4
+Requires-Dist: uvicorn>=0.30.1
 Requires-Dist: typer
 Requires-Dist: python-dotenv
-Requires-Dist: PyJWT
+Requires-Dist: PyJWT>=2.8.0
 Provides-Extra: sentry
-Requires-Dist: sentry-sdk>=2.10.0; extra == "sentry"
+Requires-Dist: sentry-sdk==2.10.0; extra == "sentry"
 Provides-Extra: firebase
-Requires-Dist: firebase-admin>=6.5.0; extra == "firebase"
-Requires-Dist: oauth2client>=4.1.3; extra == "firebase"
+Requires-Dist: firebase-admin==6.5.0; extra == "firebase"
+Requires-Dist: oauth2client==4.1.3; extra == "firebase"
 Provides-Extra: snowflakekit
 Requires-Dist: snowflakekit; extra == "snowflakekit"
 Provides-Extra: redis
-Requires-Dist: redis>=5.0.7; extra == "redis"
+Requires-Dist: redis==5.0.7; extra == "redis"
 Provides-Extra: gcloud
 Requires-Dist: google-cloud-logging; extra == "gcloud"
 

10xscale_agentflow_cli-0.1.7/10xscale_agentflow_cli.egg-info/requires.txt

@@ -0,0 +1,27 @@
+10xscale-agentflow>=0.4.0
+fastapi
+gunicorn==23.0.0
+orjson
+python-multipart>=0.0.19
+pydantic>=2.8.2
+pydantic-settings>=2.3.4
+uvicorn>=0.30.1
+typer
+python-dotenv
+PyJWT>=2.8.0
+
+[firebase]
+firebase-admin==6.5.0
+oauth2client==4.1.3
+
+[gcloud]
+google-cloud-logging
+
+[redis]
+redis==5.0.7
+
+[sentry]
+sentry-sdk==2.10.0
+
+[snowflakekit]
+snowflakekit

10xscale_agentflow_cli-0.1.7/10xscale_agentflow_cli.egg-info/top_level.txt

@@ -0,0 +1 @@
+agentflow_cli

{10xscale_agentflow_cli-0.1.6 → 10xscale_agentflow_cli-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: 10xscale-agentflow-cli
-Version: 0.1.6
+Version: 0.1.7
 Summary: CLI and API for 10xscale AgentFlow
 Author-email: 10xscale <contact@10xscale.ai>
 Maintainer-email: Shudipto Trafder <shudiptotrafder@gmail.com>
@@ -26,24 +26,24 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Requires-Dist: 10xscale-agentflow>=0.4.0
 Requires-Dist: fastapi
-Requires-Dist: gunicorn
+Requires-Dist: gunicorn==23.0.0
 Requires-Dist: orjson
-Requires-Dist: python-multipart
-Requires-Dist: pydantic
-Requires-Dist: pydantic-settings
-Requires-Dist: uvicorn
+Requires-Dist: python-multipart>=0.0.19
+Requires-Dist: pydantic>=2.8.2
+Requires-Dist: pydantic-settings>=2.3.4
+Requires-Dist: uvicorn>=0.30.1
 Requires-Dist: typer
 Requires-Dist: python-dotenv
-Requires-Dist: PyJWT
+Requires-Dist: PyJWT>=2.8.0
 Provides-Extra: sentry
-Requires-Dist: sentry-sdk>=2.10.0; extra == "sentry"
+Requires-Dist: sentry-sdk==2.10.0; extra == "sentry"
 Provides-Extra: firebase
-Requires-Dist: firebase-admin>=6.5.0; extra == "firebase"
-Requires-Dist: oauth2client>=4.1.3; extra == "firebase"
+Requires-Dist: firebase-admin==6.5.0; extra == "firebase"
+Requires-Dist: oauth2client==4.1.3; extra == "firebase"
 Provides-Extra: snowflakekit
 Requires-Dist: snowflakekit; extra == "snowflakekit"
 Provides-Extra: redis
-Requires-Dist: redis>=5.0.7; extra == "redis"
+Requires-Dist: redis==5.0.7; extra == "redis"
 Provides-Extra: gcloud
 Requires-Dist: google-cloud-logging; extra == "gcloud"
 

{10xscale_agentflow_cli-0.1.6 → 10xscale_agentflow_cli-0.1.7}/agentflow_cli/src/app/routers/graph/schemas/graph_schemas.py

@@ -37,10 +37,6 @@ class GraphInputSchema(BaseModel):
         default=ResponseGranularity.LOW,
         description="Granularity of the response (full, partial, low)",
     )
-    include_raw: bool = Field(
-        default=False,
-        description="Whether to include raw data in the response",
-    )
 
 
 class GraphInvokeOutputSchema(BaseModel):

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

@@ -5,7 +5,7 @@ from uuid import uuid4
 
 from agentflow.checkpointer import BaseCheckpointer
 from agentflow.graph import CompiledGraph
-from agentflow.state import Message
+from agentflow.state import AgentState, Message
 from agentflow.utils.thread_info import ThreadInfo
 from fastapi import BackgroundTasks, HTTPException
 from injectq import InjectQ, inject, singleton
@@ -100,38 +100,6 @@ class GraphService:
 
         return converted_messages
 
-    def _process_state_and_messages(
-        self, graph_input: GraphInputSchema, raw_state, messages: list[Message]
-    ) -> tuple[dict[str, Any] | None, list[Message]]:
-        """Process state and messages based on include_raw parameter."""
-        if graph_input.include_raw:
-            # Include everything when include_raw is True
-            state_dict = raw_state.model_dump() if raw_state is not None else raw_state
-            return state_dict, messages
-
-        # Filter out execution_meta from state and raw from messages
-        # when include_raw is False
-        if raw_state is not None:
-            state_dict = raw_state.model_dump()
-            # Remove execution_meta if present
-            if "execution_meta" in state_dict:
-                del state_dict["execution_meta"]
-        else:
-            state_dict = raw_state
-
-        # Filter raw data from messages
-        filtered_messages = []
-        for msg in messages:
-            msg_dict = msg.model_dump()
-            # Remove raw field if present
-            if "raw" in msg_dict:
-                del msg_dict["raw"]
-            # Create filtered message
-            filtered_msg = Message.model_validate(msg_dict)
-            filtered_messages.append(filtered_msg)
-
-        return state_dict, filtered_messages
-
     def _extract_context_info(
         self, raw_state, result: dict[str, Any]
     ) -> tuple[list[Message] | None, str | None]:
@@ -281,7 +249,7 @@ class GraphService:
 
             # Extract messages and state from result
             messages: list[Message] = result.get("messages", [])
-            raw_state = result.get("state", None)
+            raw_state: AgentState | None = result.get("state", None)
 
             # Extract context information using helper method
             context, context_summary = self._extract_context_info(raw_state, result)
@@ -296,14 +264,9 @@ class GraphService:
                     config["thread_id"],
                 )
 
-            # Process state and messages based on include_raw parameter
-            state_dict, processed_messages = self._process_state_and_messages(
-                graph_input, raw_state, messages
-            )
-
             return GraphInvokeOutputSchema(
-                messages=processed_messages,
-                state=state_dict,
+                messages=messages,
+                state=raw_state.model_dump() if raw_state else None,
                 context=context,
                 summary=context_summary,
                 meta=meta,

{10xscale_agentflow_cli-0.1.6 → 10xscale_agentflow_cli-0.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "10xscale-agentflow-cli"
-version = "0.1.6"
+version = "0.1.7"
 description = "CLI and API for 10xscale AgentFlow"
 readme = "README.md"
 license = {text = "MIT"}
@@ -39,15 +39,15 @@ classifiers = [
 dependencies = [
     "10xscale-agentflow>=0.4.0",
     "fastapi",
-    "gunicorn",
+    "gunicorn==23.0.0",
     "orjson",
-    "python-multipart",
-    "pydantic",
-    "pydantic-settings",
-    "uvicorn",
+    "python-multipart>=0.0.19",
+    "pydantic>=2.8.2",
+    "pydantic-settings>=2.3.4",
+    "uvicorn>=0.30.1",
     "typer",
     "python-dotenv",
-    "PyJWT",
+    "PyJWT>=2.8.0",
 ]
 
 [project.urls]
@@ -58,17 +58,17 @@ Documentation = "https://agentflow-cli.readthedocs.io/"
 
 [project.optional-dependencies]
 sentry = [
-    "sentry-sdk>=2.10.0",
+    "sentry-sdk==2.10.0",
 ]
 firebase = [
-    "firebase-admin>=6.5.0",
-    "oauth2client>=4.1.3",
+    "firebase-admin==6.5.0",
+    "oauth2client==4.1.3",
 ]
 snowflakekit = [
     "snowflakekit",
 ]
 redis = [
-    "redis>=5.0.7",
+    "redis==5.0.7",
 ]
 gcloud = [
     "google-cloud-logging",
@@ -83,7 +83,7 @@ include-package-data = true
 
 [tool.setuptools.packages.find]
 where = ["."]
-include = ["agentflow-cli*"]
+include = ["agentflow_cli*"]
 exclude = ["tests*", "docs*", "__pycache__*", "agentflow-cli/tests*"]
 
 [tool.setuptools.package-data]

10xscale_agentflow_cli-0.1.6/10xscale_agentflow_cli.egg-info/requires.txt

@@ -1,27 +0,0 @@
-10xscale-agentflow>=0.4.0
-fastapi
-gunicorn
-orjson
-python-multipart
-pydantic
-pydantic-settings
-uvicorn
-typer
-python-dotenv
-PyJWT
-
-[firebase]
-firebase-admin>=6.5.0
-oauth2client>=4.1.3
-
-[gcloud]
-google-cloud-logging
-
-[redis]
-redis>=5.0.7
-
-[sentry]
-sentry-sdk>=2.10.0
-
-[snowflakekit]
-snowflakekit

10xscale_agentflow_cli-0.1.6/10xscale_agentflow_cli.egg-info/top_level.txt

@@ -1 +0,0 @@
-