fix(agents-api): Remove broken merge artifacts

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Jan 10, 2025 · e1a1bc8 · e1a1bc8
1 parent 1a0beab
commit e1a1bc8
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 36 deletions.
diff --git a/agents-api/agents_api/activities/container.py b/agents-api/agents_api/activities/container.py
diff --git a/agents-api/agents_api/activities/execute_system.py b/agents-api/agents_api/activities/execute_system.py
@@ -32,13 +32,6 @@ async def execute_system(
 ) -> Any:
     """Execute a system call with the appropriate handler and transformed arguments."""
 
-    connection_pool = getattr(container.state, "postgres_pool", None) or getattr(
-        app.state, "postgres_pool", None
-    )
-
-    # FIXME: Remove
-    assert connection_pool is not None
-
     arguments: dict[str, Any] = system.arguments or {}
 
     connection_pool = getattr(app.state, "postgres_pool", None)

diff --git a/agents-api/agents_api/activities/task_steps/transition_step.py b/agents-api/agents_api/activities/task_steps/transition_step.py
@@ -42,8 +42,6 @@ async def transition_step(
         raise ValueError(msg)
 
     # Create transition
-    # FIXME: Remove
-    assert container.state.postgres_pool is not None
     try:
         transition = await create_execution_transition(
             developer_id=context.execution_input.developer_id,

diff --git a/agents-api/agents_api/env.py b/agents-api/agents_api/env.py
@@ -61,7 +61,7 @@
 summarization_model_name: str = env.str("SUMMARIZATION_MODEL_NAME", default="gpt-4-turbo")
 
 query_timeout: float = env.float("QUERY_TIMEOUT", default=90.0)
-pool_max_size: int = env.int("POOL_MAX_SIZE", default=multiprocessing.cpu_count() * 2)
+pool_max_size: int = env.int("POOL_MAX_SIZE", default=multiprocessing.cpu_count())
 
 
 # Auth