diff --git a/agents-api/agents_api/common/protocol/tasks.py b/agents-api/agents_api/common/protocol/tasks.py index e99db266..d3fb957b 100644 --- a/agents-api/agents_api/common/protocol/tasks.py +++ b/agents-api/agents_api/common/protocol/tasks.py @@ -29,7 +29,6 @@ WorkflowStep, ) from ...common.storage_handler import load_from_blob_store_if_remote - from .remote import BaseRemoteModel, RemoteObject # TODO: Maybe we should use a library for this diff --git a/agents-api/agents_api/common/storage_handler.py b/agents-api/agents_api/common/storage_handler.py index 73ea4c13..658b4b54 100644 --- a/agents-api/agents_api/common/storage_handler.py +++ b/agents-api/agents_api/common/storage_handler.py @@ -76,7 +76,6 @@ def load_args( new_kwargs = {} for k, v in kwargs.items(): - if isinstance(v, list): new_kwargs[k] = [ load_from_blob_store_if_remote(item) for item in v @@ -93,7 +92,6 @@ def load_args( return new_args, new_kwargs - def unload_return_value(x: Any | BaseRemoteModel | RemoteList) -> Any: if isinstance(x, (BaseRemoteModel, RemoteList)): x.unload_all()