frontend: Fix assistants_web client imports (#820)

* wip

* fix createuser issues

src/backend/config/routers.py

@@ -124,17 +124,6 @@ class RouterName(StrEnum):
             Depends(validate_organization_header),
         ],
     },
-    RouterName.DEFAULT_AGENT: {
-        "default": [
-            Depends(get_session),
-            Depends(validate_organization_header),
-        ],
-        "auth": [
-            Depends(get_session),
-            Depends(validate_authorization),
-            Depends(validate_organization_header),
-        ],
-    },
     RouterName.SNAPSHOT: {
         "default": [
             Depends(get_session),

src/backend/main.py

@@ -13,7 +13,6 @@
 )
 from backend.config.routers import ROUTER_DEPENDENCIES
 from backend.config.settings import Settings
-from backend.routers.agent import default_agent_router
 from backend.routers.agent import router as agent_router
 from backend.routers.auth import router as auth_router
 from backend.routers.chat import router as chat_router
@@ -48,7 +47,6 @@ def create_app():
         deployment_router,
         experimental_feature_router,
         agent_router,
-        default_agent_router,
         snapshot_router,
         organization_router,
         model_router,

src/backend/routers/agent.py

@@ -52,7 +52,6 @@
 )
 router.name = RouterName.AGENT
 
-
 @router.post(
     "",
     response_model=AgentPublic,
@@ -663,10 +662,3 @@ async def delete_agent_file(
     get_file_service().delete_agent_file_by_id(session, agent_id, file_id, user_id, ctx)
 
     return DeleteAgentFileResponse()
-
-
-# Default Agent Router
-default_agent_router = APIRouter(
-    prefix="/v1/default_agent",
-)
-default_agent_router.name = RouterName.DEFAULT_AGENT

src/backend/routers/chat.py

@@ -60,7 +60,6 @@ async def chat_stream(
         ctx,
     ) = process_chat(session, chat_request, request, ctx)
 
-
     return EventSourceResponse(
         generate_chat_stream(
             session,

src/backend/schemas/agent.py

@@ -1,6 +1,6 @@
 import datetime
 from enum import StrEnum
-from typing import Any, Dict, Optional
+from typing import Optional
 
 from pydantic import BaseModel, Field
 
@@ -115,16 +115,6 @@ class ListAgentsResponse(BaseModel):
     agents: list[Agent]
 
 
-class AgentTaskResponse(BaseModel):
-    task_id: str
-    status: str
-    result: Optional[Dict[str, Any]] = None
-    date_done: str
-    exception_snippet: Optional[str] = None
-    name: str
-    retries: int
-
-
 class UpdateAgentRequest(BaseModel):
     name: Optional[str] = None
     version: Optional[int] = None

src/backend/services/agent.py

@@ -6,7 +6,6 @@
 from backend.database_models.agent import Agent, AgentToolMetadata
 from backend.database_models.database import DBSessionDep
 
-TASK_TRACE_PREVIEW_LIMIT = 200
 
 def validate_agent_exists(session: DBSessionDep, agent_id: str, user_id: str) -> Agent:
     agent = agent_crud.get_agent_by_id(session, agent_id, user_id)

src/interfaces/assistants_web/src/cohere-client/client.ts

@@ -10,7 +10,7 @@ import {
   CohereUnauthorizedError,
   CreateAgentRequest,
   CreateSnapshotRequest,
-  CreateUser,
+  CreateUserV1UsersPostData,
   Fetch,
   ToggleConversationPinRequest,
   UpdateAgentRequest,
@@ -206,10 +206,8 @@ export class CohereClient {
     return this.cohereService.default.getStrategiesV1AuthStrategiesGet();
   }
 
-  public createUser(requestBody: CreateUser) {
-    return this.cohereService.default.createUserV1UsersPost({
-      requestBody,
-    });
+  public createUser(requestBody: CreateUserV1UsersPostData) {
+    return this.cohereService.default.createUserV1UsersPost(requestBody);
   }
 
   public async googleSSOAuth({ code }: { code: string }) {
@@ -271,14 +269,6 @@ export class CohereClient {
     return this.cohereService.default.getAgentByIdV1AgentsAgentIdGet({ agentId });
   }
 
-  public getAgentTasks(agentId: string) {
-    return this.cohereService.default.getAgentTasksV1AgentsAgentIdTasksGet({ agentId });
-  }
-
-  public getDefaultAgent() {
-    return this.cohereService.default.getDefaultAgentV1DefaultAgentGet();
-  }
-
   public createAgent(requestBody: CreateAgentRequest) {
     return this.cohereService.default.createAgentV1AgentsPost({ requestBody });
   }

src/interfaces/assistants_web/src/cohere-client/generated/services.gen.ts

@@ -957,7 +957,7 @@ export class DefaultService {
    * Response: Synthesized audio file.
    *
    * Raises:
-   * HTTPException: If the message with the given ID is not found.
+   * HTTPException: If the message with the given ID is not found or synthesis fails.
    * @param data The data for the request.
    * @param data.conversationId
    * @param data.messageId

src/interfaces/assistants_web/src/components/MessageRow/MessageRow.tsx

@@ -12,10 +12,9 @@ import {
   LongPressMenu,
 } from '@/components/UI';
 import { Breakpoint, useBreakpoint } from '@/hooks';
-import { useSettingsStore } from '@/stores';
 import { useExperimentalFeatures } from '@/hooks/use-experimentalFeatures';
 import { SynthesisStatus } from '@/hooks/use-synthesizer';
-
+import { useSettingsStore } from '@/stores';
 import {
   type ChatMessage,
   isAbortedMessage,
@@ -68,7 +67,7 @@ export const MessageRow = forwardRef<HTMLDivElement, Props>(function MessageRowI
   // For showing thinking steps
   const { showSteps } = useSettingsStore();
   const [isStepsExpanded, setIsStepsExpanded] = useState(true);
-      
+
   useEffect(() => {
     setIsStepsExpanded(showSteps);
   }, [showSteps]);

src/interfaces/assistants_web/src/hooks/use-agents.ts

@@ -70,35 +70,6 @@ export const useAgent = ({ agentId }: { agentId?: string }) => {
   });
 };
 
-export const useAgentTasks = ({ agentId }: { agentId?: string }) => {
-  const cohereClient = useCohereClient();
-  return useQuery({
-    queryKey: ['agentTasks', agentId],
-    queryFn: async () => {
-      try {
-        if (!agentId) {
-          throw new Error('must have agent id');
-        }
-        return await cohereClient.getAgentTasks(agentId);
-      } catch (e) {
-        console.error(e);
-        throw e;
-      }
-    },
-  });
-};
-
-export const useDefaultAgent = (enabled?: boolean) => {
-  const cohereClient = useCohereClient();
-  return useQuery({
-    queryKey: ['defaultAgent'],
-    enabled: enabled,
-    queryFn: async () => {
-      return await cohereClient.getDefaultAgent();
-    },
-  });
-};
-
 /**
  * @description Returns a function to check if an agent name is unique.
  */

src/interfaces/assistants_web/src/hooks/use-session.ts

@@ -72,9 +72,11 @@ export const useSession = () => {
   const registerMutation = useMutation({
     mutationFn: async (params: RegisterParams) => {
       return cohereClient.createUser({
-        fullname: params.name,
-        email: params.email,
-        password: params.password,
+        requestBody: {
+          fullname: params.name,
+          email: params.email,
+          password: params.password,
+        },
       });
     },
   });

src/interfaces/coral_web/src/cohere-client/client.ts

@@ -240,10 +240,6 @@ export class CohereClient {
     return this.cohereService.default.getAgentByIdV1AgentsAgentIdGet({ agentId });
   }
 
-  public getDefaultAgent() {
-    return this.cohereService.default.getDefaultAgentV1DefaultAgentGet();
-  }
-
   public createAgent(requestBody: CreateAgent) {
     return this.cohereService.default.createAgentV1AgentsPost({ requestBody });
   }

src/interfaces/coral_web/src/cohere-client/generated/services.gen.ts

@@ -39,7 +39,6 @@ import type {
   GetAgentByIdV1AgentsAgentIdGetResponse,
   GetConversationV1ConversationsConversationIdGetData,
   GetConversationV1ConversationsConversationIdGetResponse,
-  GetDefaultAgentV1DefaultAgentGetResponse,
   GetSnapshotV1SnapshotsLinkLinkIdGetData,
   GetSnapshotV1SnapshotsLinkLinkIdGetResponse,
   GetStrategiesV1AuthStrategiesGetResponse,
@@ -1274,18 +1273,6 @@ export class DefaultService {
     });
   }
 
-  /**
-   * Get Default Agent
-   * @returns GenericResponseMessage Successful Response
-   * @throws ApiError
-   */
-  public getDefaultAgentV1DefaultAgentGet(): CancelablePromise<GetDefaultAgentV1DefaultAgentGetResponse> {
-    return this.httpRequest.request({
-      method: 'GET',
-      url: '/v1/default_agent/',
-    });
-  }
-
   /**
    * List Snapshots
    * List all snapshots.

src/interfaces/coral_web/src/cohere-client/generated/types.gen.ts

@@ -847,8 +847,6 @@ export type DeleteAgentToolMetadataV1AgentsAgentIdToolMetadataAgentToolMetadataI
 export type DeleteAgentToolMetadataV1AgentsAgentIdToolMetadataAgentToolMetadataIdDeleteResponse =
   DeleteAgentToolMetadata;
 
-export type GetDefaultAgentV1DefaultAgentGetResponse = GenericResponseMessage;
-
 export type ListSnapshotsV1SnapshotsGetResponse = Array<SnapshotWithLinks>;
 
 export type CreateSnapshotV1SnapshotsPostData = {

src/interfaces/coral_web/src/hooks/agents.ts

@@ -61,17 +61,6 @@ export const useAgent = ({ agentId }: { agentId?: string }) => {
   });
 };
 
-export const useDefaultAgent = (enabled?: boolean) => {
-  const cohereClient = useCohereClient();
-  return useQuery({
-    queryKey: ['defaultAgent'],
-    enabled: enabled,
-    queryFn: async () => {
-      return await cohereClient.getDefaultAgent();
-    },
-  });
-};
-
 /**
  * @description Returns a function to check if an agent name is unique.
  */