|
@@ -1,19 +1,19 @@
|
|
|
import { get, post } from './base'
|
|
|
-import type { CustomCollectionBackend } from '@/app/components/tools/types'
|
|
|
+import type { Collection, CustomCollectionBackend, CustomParamSchema, Tool, ToolCredential } from '@/app/components/tools/types'
|
|
|
|
|
|
export const fetchCollectionList = () => {
|
|
|
- return get('/workspaces/current/tool-providers')
|
|
|
+ return get<Collection[]>('/workspaces/current/tool-providers')
|
|
|
}
|
|
|
|
|
|
export const fetchBuiltInToolList = (collectionName: string) => {
|
|
|
- return get(`/workspaces/current/tool-provider/builtin/${collectionName}/tools`)
|
|
|
+ return get<Tool[]>(`/workspaces/current/tool-provider/builtin/${collectionName}/tools`)
|
|
|
}
|
|
|
|
|
|
export const fetchCustomToolList = (collectionName: string) => {
|
|
|
- return get(`/workspaces/current/tool-provider/api/tools?provider=${collectionName}`)
|
|
|
+ return get<Tool[]>(`/workspaces/current/tool-provider/api/tools?provider=${collectionName}`)
|
|
|
}
|
|
|
export const fetchBuiltInToolCredentialSchema = (collectionName: string) => {
|
|
|
- return get(`/workspaces/current/tool-provider/builtin/${collectionName}/credentials_schema`)
|
|
|
+ return get<ToolCredential[]>(`/workspaces/current/tool-provider/builtin/${collectionName}/credentials_schema`)
|
|
|
}
|
|
|
|
|
|
export const updateBuiltInToolCredential = (collectionName: string, credential: Record<string, any>) => {
|
|
@@ -31,7 +31,7 @@ export const removeBuiltInToolCredential = (collectionName: string) => {
|
|
|
}
|
|
|
|
|
|
export const parseParamsSchema = (schema: string) => {
|
|
|
- return post('/workspaces/current/tool-provider/api/schema', {
|
|
|
+ return post<{ parameters_schema: CustomParamSchema[]; schema_type: string }>('/workspaces/current/tool-provider/api/schema', {
|
|
|
body: {
|
|
|
schema,
|
|
|
},
|
|
@@ -39,7 +39,7 @@ export const parseParamsSchema = (schema: string) => {
|
|
|
}
|
|
|
|
|
|
export const fetchCustomCollection = (collectionName: string) => {
|
|
|
- return get(`/workspaces/current/tool-provider/api/get?provider=${collectionName}`)
|
|
|
+ return get<CustomCollectionBackend>(`/workspaces/current/tool-provider/api/get?provider=${collectionName}`)
|
|
|
}
|
|
|
|
|
|
export const createCustomCollection = (collection: CustomCollectionBackend) => {
|