fix(agents-api): Lint and fix sample task

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Nov 29, 2024 · 3a32c7c · 3a32c7c
1 parent 9ce2d6b
commit 3a32c7c
Show file tree

Hide file tree

Showing 5 changed files with 2 additions and 10 deletions.
diff --git a/agents-api/agents_api/activities/execute_system.py b/agents-api/agents_api/activities/execute_system.py
@@ -18,7 +18,6 @@
 )
 from ..autogen.Sessions import CreateSessionRequest
 from ..autogen.Tools import SystemDef
-from ..common.protocol.remote import RemoteObject
 from ..common.protocol.tasks import StepContext
 from ..common.storage_handler import auto_blob_store, load_from_blob_store_if_remote
 from ..env import testing

diff --git a/agents-api/agents_api/activities/task_steps/prompt_step.py b/agents-api/agents_api/activities/task_steps/prompt_step.py
@@ -1,9 +1,5 @@
-from typing import Callable
 
-from anthropic.types.beta.beta_message import BetaMessage
 from beartype import beartype
-from langchain_core.tools import BaseTool
-from langchain_core.tools.convert import tool as tool_decorator
 from litellm.types.utils import ModelResponse
 from temporalio import activity
 from temporalio.exceptions import ApplicationError
@@ -16,7 +12,6 @@
 from ...common.storage_handler import auto_blob_store
 from ...common.utils.template import render_template
 from ...env import debug
-from ..utils import get_handler_with_filtered_params
 from .base_evaluate import base_evaluate
 
 COMPUTER_USE_BETA_FLAG = "computer-use-2024-10-22"

diff --git a/agents-api/agents_api/autogen/Tasks.py b/agents-api/agents_api/autogen/Tasks.py
@@ -9,7 +9,6 @@
 from pydantic import AwareDatetime, BaseModel, ConfigDict, Field, StrictBool
 
 from .Chat import ChatSettings
-from .Common import JinjaTemplate
 from .Tools import (
     ChosenBash20241022,
     ChosenComputer20241022,

diff --git a/agents-api/agents_api/routers/sessions/chat.py b/agents-api/agents_api/routers/sessions/chat.py
@@ -1,5 +1,4 @@
-from datetime import datetime
-from typing import Annotated, Callable, Optional
+from typing import Annotated, Optional
 from uuid import UUID, uuid4
 
 from fastapi import BackgroundTasks, Depends, Header, HTTPException, status

diff --git a/agents-api/tests/sample_tasks/simple_multi_step.yaml b/agents-api/tests/sample_tasks/simple_multi_step.yaml
@@ -20,4 +20,4 @@ main:
   - log: _
 
   - evaluate:
-      result: _["choices"][0]["message"].content.strip()
+      result: '[item["choices"][0]["message"].content.strip() for item in _]'