address comments left by @creatorrr on #986 (feat(agents-api): Configure routes );

Author: ellipsis-dev[bot]

Diff for: agents-api/agents_api/activities/task_steps/__init__.py

@@ -2,7 +2,6 @@
 
 from .base_evaluate import base_evaluate
 
-# from .cozo_query_step import cozo_query_step
 from .evaluate_step import evaluate_step
 from .for_each_step import for_each_step
 from .get_value_step import get_value_step

Diff for: agents-api/agents_api/env.py

@@ -49,8 +49,6 @@
 
 # Cozo
 # ----
-cozo_host: str = env.str("COZO_HOST", default="http://127.0.0.1:9070")
-cozo_auth: str = env.str("COZO_AUTH_TOKEN", default=None)
 summarization_model_name: str = env.str(
     "SUMMARIZATION_MODEL_NAME", default="gpt-4-turbo"
 )

Diff for: agents-api/agents_api/queries/executions/create_execution.py

@@ -14,7 +14,6 @@
     rewrap_exceptions,
     wrap_in_class,
 )
-from .constants import OUTPUT_UNNEST_KEY
 
 ModelT = TypeVar("ModelT", bound=Any)
 T = TypeVar("T")
@@ -33,7 +32,6 @@
 #     transform=lambda d: {"id": d["execution_id"], **d},
 #     _kind="inserted",
 # )
-# @cozo_query
 # @increase_counter("create_execution")
 # @beartype
 async def create_execution(

Diff for: agents-api/agents_api/queries/executions/get_execution.py

@@ -9,7 +9,6 @@
     pg_query,
     wrap_in_class,
 )
-from .constants import OUTPUT_UNNEST_KEY
 
 ModelT = TypeVar("ModelT", bound=Any)
 T = TypeVar("T")
@@ -27,6 +26,7 @@
 #         AssertionError: partialclass(HTTPException, status_code=404),
 #         QueryException: partialclass(HTTPException, status_code=400),
 #         ValidationError: partialclass(HTTPException, status_code=400),
+
 #         TypeError: partialclass(HTTPException, status_code=400),
 #     }
 # )

Diff for: agents-api/agents_api/queries/executions/list_execution_transitions.py

@@ -7,6 +7,7 @@
 from pydantic import ValidationError
 
 from ...autogen.openapi_model import Transition
+
 from ..utils import cozo_query, partialclass, rewrap_exceptions, wrap_in_class
 
 ModelT = TypeVar("ModelT", bound=Any)
@@ -21,7 +22,6 @@
     }
 )
 @wrap_in_class(Transition)
-@cozo_query
 @beartype
 async def list_execution_transitions(
     *,

Diff for: agents-api/agents_api/queries/executions/list_executions.py

@@ -0,0 +1 @@
+[the existing file is empty]

Diff for: agents-api/agents_api/queries/executions/prepare_execution_input.py

@@ -53,7 +53,6 @@
         ],
     },
 )
-@cozo_query
 @beartype
 async def prepare_execution_input(
     *,

Diff for: agents-api/agents_api/queries/executions/update_execution.py

@@ -23,7 +23,6 @@
     verify_developer_owns_resource_query,
     wrap_in_class,
 )
-from .constants import OUTPUT_UNNEST_KEY
 
 ModelT = TypeVar("ModelT", bound=Any)
 T = TypeVar("T")
@@ -42,7 +41,6 @@
     transform=lambda d: {"id": d["execution_id"], **d},
     _kind="inserted",
 )
-@cozo_query
 @increase_counter("update_execution")
 @beartype
 async def update_execution(