11import { MySchema } from "@/schemas/MySchema" ;
22import { ExtractedData } from "llama-cloud-services/beta/agent" ;
3- import { ApiClients } from "@llamaindex/ui" ;
43import {
5- createCloudAgentClient ,
6- createLlamaDeployClient ,
7- createLlamaDeployConfig ,
8- cloudApiClient ,
4+ ApiClients ,
5+ createWorkflowClient ,
6+ createWorkflowConfig ,
97} from "@llamaindex/ui" ;
10- import { EXTRACTED_DATA_COLLECTION } from "./config" ;
8+ import { createCloudAgentClient , cloudApiClient } from "@llamaindex/ui" ;
9+ import { AGENT_NAME , EXTRACTED_DATA_COLLECTION } from "./config" ;
1110
1211const platformToken = import . meta. env . VITE_LLAMA_CLOUD_API_KEY ;
1312const apiBaseUrl = import . meta. env . VITE_LLAMA_CLOUD_BASE_URL ;
@@ -31,8 +30,14 @@ const agentClient = createCloudAgentClient<ExtractedData<MySchema>>({
3130 collection : EXTRACTED_DATA_COLLECTION ,
3231} ) ;
3332
33+ const workflowsClient = createWorkflowClient (
34+ createWorkflowConfig ( {
35+ baseUrl : `/deployments/${ AGENT_NAME } /` ,
36+ } ) ,
37+ ) ;
38+
3439const clients : ApiClients = {
35- llamaDeployClient : createLlamaDeployClient ( createLlamaDeployConfig ( ) ) ,
40+ workflowsClient : workflowsClient ,
3641 cloudApiClient : cloudApiClient ,
3742 agentDataClient : agentClient ,
3843} ;
0 commit comments