diff --git a/fastagency_studio/studio/helpers.py b/fastagency_studio/studio/helpers.py index a3184d4c..00c88fcb 100644 --- a/fastagency_studio/studio/helpers.py +++ b/fastagency_studio/studio/helpers.py @@ -4,13 +4,12 @@ from uuid import UUID from asyncer import asyncify -from fastapi import BackgroundTasks, HTTPException - from fastagency.studio.saas_app_generator import ( InvalidFlyTokenError, InvalidGHTokenError, SaasAppGenerator, ) +from fastapi import BackgroundTasks, HTTPException from .auth_token.auth import create_deployment_auth_token from .db.base import DefaultDB diff --git a/fastagency_studio/studio/models/agents/base.py b/fastagency_studio/studio/models/agents/base.py index ce4118d4..07aca81b 100644 --- a/fastagency_studio/studio/models/agents/base.py +++ b/fastagency_studio/studio/models/agents/base.py @@ -1,9 +1,8 @@ from typing import Annotated, Optional, Union from uuid import UUID -from typing_extensions import TypeAlias - from fastagency.api.openapi import OpenAPI +from typing_extensions import TypeAlias from ..base import Field, Model from ..registry import Registry diff --git a/fastagency_studio/studio/models/agents/user_proxy.py b/fastagency_studio/studio/models/agents/user_proxy.py index a4af004c..0ccef538 100644 --- a/fastagency_studio/studio/models/agents/user_proxy.py +++ b/fastagency_studio/studio/models/agents/user_proxy.py @@ -2,8 +2,8 @@ from uuid import UUID import autogen - from fastagency.api.openapi.client import OpenAPI + from ..base import Field, Model from ..registry import register diff --git a/fastagency_studio/studio/models/agents/web_surfer.py b/fastagency_studio/studio/models/agents/web_surfer.py index 5d619ea7..aabc2a01 100644 --- a/fastagency_studio/studio/models/agents/web_surfer.py +++ b/fastagency_studio/studio/models/agents/web_surfer.py @@ -4,9 +4,8 @@ from asyncer import syncify from autogen.agentchat import AssistantAgent as AutoGenAssistantAgent from autogen.agentchat import ConversableAgent as AutoGenConversableAgent -from typing_extensions import TypeAlias - from fastagency.studio.models.agents.web_surfer_autogen import WebSurferChat +from typing_extensions import TypeAlias from ..base import Field, Model from ..registry import register @@ -47,7 +46,7 @@ def create_new_task( task: Annotated[str, "task for websurfer"], ) -> str: try: - return syncify(self.websurfer_chat.create_new_task)(task) + return syncify(self.websurfer_chat.create_new_task)(task) # type: ignore [no-any-return] except Exception as e: raise e @@ -62,7 +61,7 @@ def continue_task_with_additional_instructions( ], ) -> str: try: - return syncify( + return syncify( # type: ignore [no-any-return] self.websurfer_chat.continue_task_with_additional_instructions )(message) except Exception as e: diff --git a/fastagency_studio/studio/models/toolboxes/toolbox.py b/fastagency_studio/studio/models/toolboxes/toolbox.py index fe080b11..048b4479 100644 --- a/fastagency_studio/studio/models/toolboxes/toolbox.py +++ b/fastagency_studio/studio/models/toolboxes/toolbox.py @@ -2,10 +2,10 @@ from uuid import UUID import httpx +from fastagency.api.openapi.client import OpenAPI from pydantic import AfterValidator, HttpUrl from typing_extensions import TypeAlias -from fastagency.api.openapi.client import OpenAPI from ..base import Field, Model from ..registry import Registry