macrocosmos 1.2.18 → 1.2.20

package/README.md

@@ -12,7 +12,7 @@ npm install macrocosmos
 
 ### Apex Client
 
-The Apex client provides an interface for accessing the Apex API for chat completions and web search.
+The Apex client provides an interface for accessing the Apex API for chat completions, web search and deep research.
 
 ```typescript
 import { ApexClient } from 'macrocosmos';
@@ -47,7 +47,7 @@ const submittedJob = await client.submitDeepResearcherJob({
     });
 
 // Get the results of a deep research job
-const polledJobResult = await client.getDeepResearcherJob(submittedJob.jobId);
+const polledJobResult = await client.getDeepResearcherJob({jobId: submittedJob.jobId});
 ```
 
 ### Gravity Client

package/dist/README.md

@@ -12,7 +12,7 @@ npm install macrocosmos
 
 ### Apex Client
 
-The Apex client provides an interface for accessing the Apex API for chat completions and web search.
+The Apex client provides an interface for accessing the Apex API for chat completions, web search and deep research.
 
 ```typescript
 import { ApexClient } from 'macrocosmos';
@@ -47,7 +47,7 @@ const submittedJob = await client.submitDeepResearcherJob({
     });
 
 // Get the results of a deep research job
-const polledJobResult = await client.getDeepResearcherJob(submittedJob.jobId);
+const polledJobResult = await client.getDeepResearcherJob({jobId: submittedJob.jobId});
 ```
 
 ### Gravity Client

package/dist/__tests__/apex/client-chat.test.js

@@ -65,11 +65,6 @@ describe("ApexClient", () => {
         expect(response.choices?.[0]?.message?.content?.toLowerCase()).toContain("paris");
     }, 30000); // Increase timeout to 30 seconds
     it("should retrieve stored chat completions", async () => {
-        // Create ApexClient
-        const client = new macrocosmos_1.ApexClient({
-            apiKey: API_KEY,
-            appName: "apex-client.test.ts",
-        });
         // chat ID for testing
         const chatId = "test-chat-id";
         // Get stored chat completions
@@ -81,13 +76,44 @@ describe("ApexClient", () => {
         // chat id doesn't exist so check first element is an empty object
         expect(Object.keys(result.chatCompletions[0] || {}).length).toBe(0);
     }, 30000);
+    // it("should append a chat completion", async () => {
+    //   // chat ID for testing
+    //
+    //   const create_chat_result = await client.createChatAndCompletion({
+    //     userPrompt: "This is a test chat, how are you?",
+    //     chatType: "apex",
+    //     completionType: "basic",
+    //     title: "Test Chat",
+    //   });
+    //
+    //   // Verify the response structure
+    //   console.log("Create chat and completion response:", create_chat_result);
+    //   expect(create_chat_result).toBeDefined();
+    //   // Get stored chat completions
+    //   const result = await client.createCompletion({
+    //     chatId: create_chat_result.parsedChat?.id ?? "",
+    //     userPrompt: "This is a test completion, how are you?",
+    //     completionType: "basic",
+    //   });
+    //
+    //   // Verify the response structure
+    //   console.log("Create completion response:", result);
+    //   expect(result).toBeDefined();
+    //
+    //   // Check ParsedCompletion
+    //   expect(result.parsedCompletion).toBeDefined();
+    //   expect(result.parsedCompletion!.id).toBeDefined();
+    //   expect(result.parsedCompletion!.chatId).toBe(
+    //     create_chat_result.parsedChat!.id,
+    //   );
+    //   expect(result.parsedCompletion!.completionType).toBe("basic");
+    //   expect(result.parsedCompletion!.userPromptText).toBe(
+    //     "This is a test completion, how are you?",
+    //   );
+    //
+    // }, 30000);
     // Deep Researcher Tests
     it("should create a deep research job", async () => {
-        // Create ApexClient
-        const client = new macrocosmos_1.ApexClient({
-            apiKey: API_KEY,
-            appName: "apex-client.test.ts",
-        });
         // Create test parameters
         const params = {
             messages: deepResearchMessages,
@@ -108,11 +134,6 @@ describe("ApexClient", () => {
         console.log("Create Job Response:", response);
     }, 30000);
     it("should get deep researcher job results", async () => {
-        // Create ApexClient
-        const client = new macrocosmos_1.ApexClient({
-            apiKey: API_KEY,
-            appName: "apex-client.test.ts",
-        });
         // First create a job. Use simple inputs as job creation is tested in the previous test.
         const createParams = {
             messages: deepResearchMessages,
@@ -120,7 +141,7 @@ describe("ApexClient", () => {
         const createResponse = await client.submitDeepResearcherJob(createParams);
         const jobId = createResponse.jobId;
         // Then get the results
-        const response = await client.getDeepResearcherJob({ jobId });
+        const response = await client.getDeepResearcherJob({ jobId: jobId });
         // Verify response structure
         expect(response).toBeDefined();
         expect(response.jobId).toBe(jobId); // Job ID should match the one from the create response
@@ -132,4 +153,55 @@ describe("ApexClient", () => {
         // Log response for debugging
         console.log("Get Job Results Response:", response);
     }, 60000); // Longer timeout for this test as it involves multiple API calls
+    // TODO: Uncomment this when delete chat is available, so we can remove all test chats first
+    // it("should retrieve a user's completions based on the search term", async () => {
+    //   const searchTerm = "France?";
+    //   // Get stored chat completions
+    //   const result = await client.searchChatIdsByPromptAndCompletionText({
+    //     searchTerm: searchTerm,
+    //   });
+    //   // Verify the response structure
+    //   console.log("Stored chats:", result);
+    //   expect(result).toBeDefined();
+    //   expect(Array.isArray(result.chatIds)).toBe(true);
+    //   // chat id doesn't exist so check first element is an empty object
+    //   expect(Object.keys(result.chatIds[0] || {}).length).toBe(0);
+    // }, 30000);
+    // it("should create a chat and completion for a user", async () => {
+    //   const result = await client.createChatAndCompletion({
+    //     userPrompt: "This is a test chat, how are you?",
+    //     chatType: "apex",
+    //     completionType: "basic",
+    //     title: "Test Chat",
+    //   });
+    //   // Verify the response structure
+    //   console.log("Create chat and completion response:", result);
+    //   expect(result).toBeDefined();
+    //   // Check ParsedChat
+    //   expect(result.parsedChat).toBeDefined();
+    //   expect(result.parsedChat!.id).toBeDefined();
+    //   expect(result.parsedChat!.title).toBe("Test Chat");
+    //   expect(result.parsedChat!.chatType).toBe("apex");
+    //   expect(result.parsedChat!.createdAt).toBeDefined();
+    //   // Check ParsedCompletion
+    //   expect(result.parsedCompletion).toBeDefined();
+    //   expect(result.parsedCompletion!.id).toBeDefined();
+    //   expect(result.parsedCompletion!.chatId).toBe(result.parsedChat!.id);
+    //   expect(result.parsedCompletion!.completionType).toBe("basic");
+    //   expect(result.parsedCompletion!.userPromptText).toBe(
+    //     "This is a test chat, how are you?",
+    //   );
+    //   expect(result.parsedCompletion!.completionText).toBeDefined();
+    //   expect(result.parsedCompletion!.createdAt).toBeDefined();
+    // }, 30000);
+    // it("should retrieve a user's stored chats", async () => {
+    //   // Get stored chat completions
+    //   const result = await client.getChatSessions();
+    //   // Verify the response structure
+    //   console.log("Stored chats:", result);
+    //   expect(result).toBeDefined();
+    //   expect(Array.isArray(result.chatSessions)).toBe(true);
+    //   // chat id doesn't exist so check first element is an empty object
+    //   expect(Object.keys(result.chatSessions[0] || {}).length).toBe(0);
+    // }, 30000);
 });

package/dist/__tests__/gravity/client.test.js

@@ -50,7 +50,7 @@ describe("GravityClient", () => {
         it("should call the CreateGravityTask method on the gRPC client", async () => {
             const params = {
                 gravityTasks: [{ platform: "x", topic: "#ai" }],
-                name: "Test Task",
+                name: "Test Task from typescript SDK",
             };
             const result = await client.createGravityTask(params);
             expect(mockGrpcClient.createGravityTask).toHaveBeenCalledWith(params, expect.any(Function));

package/dist/generated/apex/v1/apex.d.ts

@@ -462,7 +462,7 @@ export interface GetStoredChatCompletionsResponse {
     chatCompletions: StoredChatCompletion[];
 }
 /** Directly model the attributes as a map */
-export interface UpdateChatAttributeRequest {
+export interface UpdateChatAttributesRequest {
     /** chat_id: the unique id associated to a users chat message */
     chatId: string;
     /** attributes: the data attributes captured in the chat logging process */
@@ -470,14 +470,14 @@ export interface UpdateChatAttributeRequest {
         [key: string]: string;
     };
 }
-export interface UpdateChatAttributeRequest_AttributesEntry {
+export interface UpdateChatAttributesRequest_AttributesEntry {
     key: string;
     value: string;
 }
-/** A UpdateChatAttribute response */
-export interface UpdateChatAttributeResponse {
-    /** success: indicates if an attribute update was successful */
-    success: boolean;
+/** A UpdateChatAttributes response */
+export interface UpdateChatAttributesResponse {
+    /** chat: the updated chat row from the chats table */
+    chat?: ChatSession | undefined;
 }
 /** A DeleteChats request */
 export interface DeleteChatsRequest {
@@ -489,6 +489,109 @@ export interface DeleteChatsResponse {
     /** success: indicates if the deletion was successful */
     success: boolean;
 }
+/** A ParsedChat message serving as part of the CreateChatAndCompletion response */
+export interface ParsedChat {
+    /** id: the chat_id */
+    id: string;
+    /** title: title of the chat */
+    title: string;
+    /** created_at: the time the chat was created */
+    createdAt?: Date | undefined;
+    /** chat_type: the service a single chat can be using (e.g.: "apex", "gravity") */
+    chatType: string;
+}
+/** A ParsedCompletion message serving as part of the CreateChatAndCompletion response */
+export interface ParsedCompletion {
+    /** id: the completion_id */
+    id: string;
+    /** chat_id: the chat_id */
+    chatId: string;
+    /** created_at: the time the completion was created */
+    createdAt?: Date | undefined;
+    /** user_prompt_text: the user's chat prompt text */
+    userPromptText: string;
+    /** completion_text: the user's completion text */
+    completionText: string;
+    /** completion_type: type of completion */
+    completionType: string;
+    /** metadata: metadata json blob */
+    metadata?: {
+        [key: string]: any;
+    } | undefined;
+}
+/** A CreateChatAndCompletion request */
+export interface CreateChatAndCompletionRequest {
+    /** user_prompt: the prompt the user issues */
+    userPrompt: string;
+    /** chat_type: the service a single chat can be using (e.g.: "apex", "gravity") */
+    chatType: string;
+    /** completion_type: specific to completions and might accompany specific kinds of metadata (e.g.: "basic", "combined") */
+    completionType: string;
+    /** title: the title of the new chat (optional) */
+    title?: string | undefined;
+}
+/** A CreateChatAndCompletion response */
+export interface CreateChatAndCompletionResponse {
+    /** parsed_chat: the chat row that was successfully created */
+    parsedChat?: ParsedChat | undefined;
+    /** parsed_completion: the completion row that was successfully created */
+    parsedCompletion?: ParsedCompletion | undefined;
+}
+/** A CreateCompletion request */
+export interface CreateCompletionRequest {
+    /** chat_id: the ID of the chat to create the completion for */
+    chatId: string;
+    /** user_prompt: the prompt the user issues */
+    userPrompt: string;
+    /** completion_type: the completion type e.g. basic, reasoning etc. */
+    completionType: string;
+}
+/** A CreateCompletion response */
+export interface CreateCompletionResponse {
+    /** parsed_completion: the completion row that was successfully created */
+    parsedCompletion?: ParsedCompletion | undefined;
+}
+/** A DeleteCompletions request */
+export interface DeleteCompletionsRequest {
+    /** completion_ids: the unique ids associated to user chat-completions that should be deleted */
+    completionIds: string[];
+}
+/** A DeleteCompletions response */
+export interface DeleteCompletionsResponse {
+    /** success: indicates if the deletion was successful */
+    success: boolean;
+}
+/** A SearchChatIdsByPromptAndCompletionText request */
+export interface SearchChatIdsByPromptAndCompletionTextRequest {
+    /** search_term: the user supplied search term */
+    searchTerm: string;
+}
+/** A SearchChatIdsByPromptAndCompletionText response */
+export interface SearchChatIdsByPromptAndCompletionTextResponse {
+    /** chat_ids: a list of chat_ids which adhere to the search criteria */
+    chatIds: string[];
+}
+/** An UpdateCompletionAttributes request */
+export interface UpdateCompletionAttributesRequest {
+    /** completion_id: the chat completion id */
+    completionId: string;
+    /** completion_text: the user's completion text (optional) */
+    completionText?: string | undefined;
+    /** metadata: metadata json blob (optional) */
+    metadata?: {
+        [key: string]: any;
+    } | undefined;
+}
+/** An UpdateCompletionAttributes response */
+export interface UpdateCompletionAttributesResponse {
+    /** completion: the chat completion that has been updated */
+    completion?: StoredChatCompletion | undefined;
+}
+/** A GetCompletionsWithDeepResearcherEntry response (request is not required) */
+export interface GetCompletionsWithDeepResearcherEntryResponse {
+    /** completions: a list of completion objects containing deep researcher metadata */
+    completions: ParsedCompletion[];
+}
 export declare const ChatCompletionRequest: MessageFns<ChatCompletionRequest>;
 export declare const SamplingParameters: MessageFns<SamplingParameters>;
 export declare const ChatCompletionResponse: MessageFns<ChatCompletionResponse>;
@@ -524,11 +627,24 @@ export declare const GetChatSessionsResponse: MessageFns<GetChatSessionsResponse
 export declare const GetStoredChatCompletionsRequest: MessageFns<GetStoredChatCompletionsRequest>;
 export declare const StoredChatCompletion: MessageFns<StoredChatCompletion>;
 export declare const GetStoredChatCompletionsResponse: MessageFns<GetStoredChatCompletionsResponse>;
-export declare const UpdateChatAttributeRequest: MessageFns<UpdateChatAttributeRequest>;
-export declare const UpdateChatAttributeRequest_AttributesEntry: MessageFns<UpdateChatAttributeRequest_AttributesEntry>;
-export declare const UpdateChatAttributeResponse: MessageFns<UpdateChatAttributeResponse>;
+export declare const UpdateChatAttributesRequest: MessageFns<UpdateChatAttributesRequest>;
+export declare const UpdateChatAttributesRequest_AttributesEntry: MessageFns<UpdateChatAttributesRequest_AttributesEntry>;
+export declare const UpdateChatAttributesResponse: MessageFns<UpdateChatAttributesResponse>;
 export declare const DeleteChatsRequest: MessageFns<DeleteChatsRequest>;
 export declare const DeleteChatsResponse: MessageFns<DeleteChatsResponse>;
+export declare const ParsedChat: MessageFns<ParsedChat>;
+export declare const ParsedCompletion: MessageFns<ParsedCompletion>;
+export declare const CreateChatAndCompletionRequest: MessageFns<CreateChatAndCompletionRequest>;
+export declare const CreateChatAndCompletionResponse: MessageFns<CreateChatAndCompletionResponse>;
+export declare const CreateCompletionRequest: MessageFns<CreateCompletionRequest>;
+export declare const CreateCompletionResponse: MessageFns<CreateCompletionResponse>;
+export declare const DeleteCompletionsRequest: MessageFns<DeleteCompletionsRequest>;
+export declare const DeleteCompletionsResponse: MessageFns<DeleteCompletionsResponse>;
+export declare const SearchChatIdsByPromptAndCompletionTextRequest: MessageFns<SearchChatIdsByPromptAndCompletionTextRequest>;
+export declare const SearchChatIdsByPromptAndCompletionTextResponse: MessageFns<SearchChatIdsByPromptAndCompletionTextResponse>;
+export declare const UpdateCompletionAttributesRequest: MessageFns<UpdateCompletionAttributesRequest>;
+export declare const UpdateCompletionAttributesResponse: MessageFns<UpdateCompletionAttributesResponse>;
+export declare const GetCompletionsWithDeepResearcherEntryResponse: MessageFns<GetCompletionsWithDeepResearcherEntryResponse>;
 export type ApexServiceService = typeof ApexServiceService;
 export declare const ApexServiceService: {
     /** ChatCompletion generates a completion for a given request. */
@@ -601,15 +717,15 @@ export declare const ApexServiceService: {
         readonly responseSerialize: (value: GetStoredChatCompletionsResponse) => Buffer<ArrayBuffer>;
         readonly responseDeserialize: (value: Buffer) => GetStoredChatCompletionsResponse;
     };
-    /** UpdateChatAttribute updates attribute after asking LLM for one */
-    readonly updateChatAttribute: {
-        readonly path: "/apex.v1.ApexService/UpdateChatAttribute";
+    /** UpdateChatAttributes updates specified attributes of a chat */
+    readonly updateChatAttributes: {
+        readonly path: "/apex.v1.ApexService/UpdateChatAttributes";
         readonly requestStream: false;
         readonly responseStream: false;
-        readonly requestSerialize: (value: UpdateChatAttributeRequest) => Buffer<ArrayBuffer>;
-        readonly requestDeserialize: (value: Buffer) => UpdateChatAttributeRequest;
-        readonly responseSerialize: (value: UpdateChatAttributeResponse) => Buffer<ArrayBuffer>;
-        readonly responseDeserialize: (value: Buffer) => UpdateChatAttributeResponse;
+        readonly requestSerialize: (value: UpdateChatAttributesRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => UpdateChatAttributesRequest;
+        readonly responseSerialize: (value: UpdateChatAttributesResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => UpdateChatAttributesResponse;
     };
     /** DeleteChats removes chats based on the specified chat_ids */
     readonly deleteChats: {
@@ -621,6 +737,66 @@ export declare const ApexServiceService: {
         readonly responseSerialize: (value: DeleteChatsResponse) => Buffer<ArrayBuffer>;
         readonly responseDeserialize: (value: Buffer) => DeleteChatsResponse;
     };
+    /** CreateChatAndCompletion makes a new chat when completion present, otherwise sets as new chat */
+    readonly createChatAndCompletion: {
+        readonly path: "/apex.v1.ApexService/CreateChatAndCompletion";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: CreateChatAndCompletionRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => CreateChatAndCompletionRequest;
+        readonly responseSerialize: (value: CreateChatAndCompletionResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => CreateChatAndCompletionResponse;
+    };
+    /** CreateCompletion creates a new completion for an existing chat */
+    readonly createCompletion: {
+        readonly path: "/apex.v1.ApexService/CreateCompletion";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: CreateCompletionRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => CreateCompletionRequest;
+        readonly responseSerialize: (value: CreateCompletionResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => CreateCompletionResponse;
+    };
+    /** DeleteCompletions removes completions based on the specified completion_ids */
+    readonly deleteCompletions: {
+        readonly path: "/apex.v1.ApexService/DeleteCompletions";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: DeleteCompletionsRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => DeleteCompletionsRequest;
+        readonly responseSerialize: (value: DeleteCompletionsResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => DeleteCompletionsResponse;
+    };
+    /** SearchChatIdsByPromptAndCompletionText searches for completions where the parent chat has the user's ID and either user_prompt_text OR completion_text matches the search_term */
+    readonly searchChatIdsByPromptAndCompletionText: {
+        readonly path: "/apex.v1.ApexService/SearchChatIdsByPromptAndCompletionText";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: SearchChatIdsByPromptAndCompletionTextRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => SearchChatIdsByPromptAndCompletionTextRequest;
+        readonly responseSerialize: (value: SearchChatIdsByPromptAndCompletionTextResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => SearchChatIdsByPromptAndCompletionTextResponse;
+    };
+    /** UpdateCompletionAttributes updates attribute for a given chat completion */
+    readonly updateCompletionAttributes: {
+        readonly path: "/apex.v1.ApexService/UpdateCompletionAttributes";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: UpdateCompletionAttributesRequest) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => UpdateCompletionAttributesRequest;
+        readonly responseSerialize: (value: UpdateCompletionAttributesResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => UpdateCompletionAttributesResponse;
+    };
+    /** GetCompletionsWithDeepResearcherEntry returns a list of completions that contain metadata on deep researcher */
+    readonly getCompletionsWithDeepResearcherEntry: {
+        readonly path: "/apex.v1.ApexService/GetCompletionsWithDeepResearcherEntry";
+        readonly requestStream: false;
+        readonly responseStream: false;
+        readonly requestSerialize: (value: Empty) => Buffer<ArrayBuffer>;
+        readonly requestDeserialize: (value: Buffer) => Empty;
+        readonly responseSerialize: (value: GetCompletionsWithDeepResearcherEntryResponse) => Buffer<ArrayBuffer>;
+        readonly responseDeserialize: (value: Buffer) => GetCompletionsWithDeepResearcherEntryResponse;
+    };
 };
 export interface ApexServiceServer extends UntypedServiceImplementation {
     /** ChatCompletion generates a completion for a given request. */
@@ -637,10 +813,22 @@ export interface ApexServiceServer extends UntypedServiceImplementation {
     getChatSessions: handleUnaryCall<Empty, GetChatSessionsResponse>;
     /** GetStoredChatCompletions retrieves a chat's completions */
     getStoredChatCompletions: handleUnaryCall<GetStoredChatCompletionsRequest, GetStoredChatCompletionsResponse>;
-    /** UpdateChatAttribute updates attribute after asking LLM for one */
-    updateChatAttribute: handleUnaryCall<UpdateChatAttributeRequest, UpdateChatAttributeResponse>;
+    /** UpdateChatAttributes updates specified attributes of a chat */
+    updateChatAttributes: handleUnaryCall<UpdateChatAttributesRequest, UpdateChatAttributesResponse>;
     /** DeleteChats removes chats based on the specified chat_ids */
     deleteChats: handleUnaryCall<DeleteChatsRequest, DeleteChatsResponse>;
+    /** CreateChatAndCompletion makes a new chat when completion present, otherwise sets as new chat */
+    createChatAndCompletion: handleUnaryCall<CreateChatAndCompletionRequest, CreateChatAndCompletionResponse>;
+    /** CreateCompletion creates a new completion for an existing chat */
+    createCompletion: handleUnaryCall<CreateCompletionRequest, CreateCompletionResponse>;
+    /** DeleteCompletions removes completions based on the specified completion_ids */
+    deleteCompletions: handleUnaryCall<DeleteCompletionsRequest, DeleteCompletionsResponse>;
+    /** SearchChatIdsByPromptAndCompletionText searches for completions where the parent chat has the user's ID and either user_prompt_text OR completion_text matches the search_term */
+    searchChatIdsByPromptAndCompletionText: handleUnaryCall<SearchChatIdsByPromptAndCompletionTextRequest, SearchChatIdsByPromptAndCompletionTextResponse>;
+    /** UpdateCompletionAttributes updates attribute for a given chat completion */
+    updateCompletionAttributes: handleUnaryCall<UpdateCompletionAttributesRequest, UpdateCompletionAttributesResponse>;
+    /** GetCompletionsWithDeepResearcherEntry returns a list of completions that contain metadata on deep researcher */
+    getCompletionsWithDeepResearcherEntry: handleUnaryCall<Empty, GetCompletionsWithDeepResearcherEntryResponse>;
 }
 export interface ApexServiceClient extends Client {
     /** ChatCompletion generates a completion for a given request. */
@@ -670,14 +858,38 @@ export interface ApexServiceClient extends Client {
     getStoredChatCompletions(request: GetStoredChatCompletionsRequest, callback: (error: ServiceError | null, response: GetStoredChatCompletionsResponse) => void): ClientUnaryCall;
     getStoredChatCompletions(request: GetStoredChatCompletionsRequest, metadata: Metadata, callback: (error: ServiceError | null, response: GetStoredChatCompletionsResponse) => void): ClientUnaryCall;
     getStoredChatCompletions(request: GetStoredChatCompletionsRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: GetStoredChatCompletionsResponse) => void): ClientUnaryCall;
-    /** UpdateChatAttribute updates attribute after asking LLM for one */
-    updateChatAttribute(request: UpdateChatAttributeRequest, callback: (error: ServiceError | null, response: UpdateChatAttributeResponse) => void): ClientUnaryCall;
-    updateChatAttribute(request: UpdateChatAttributeRequest, metadata: Metadata, callback: (error: ServiceError | null, response: UpdateChatAttributeResponse) => void): ClientUnaryCall;
-    updateChatAttribute(request: UpdateChatAttributeRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: UpdateChatAttributeResponse) => void): ClientUnaryCall;
+    /** UpdateChatAttributes updates specified attributes of a chat */
+    updateChatAttributes(request: UpdateChatAttributesRequest, callback: (error: ServiceError | null, response: UpdateChatAttributesResponse) => void): ClientUnaryCall;
+    updateChatAttributes(request: UpdateChatAttributesRequest, metadata: Metadata, callback: (error: ServiceError | null, response: UpdateChatAttributesResponse) => void): ClientUnaryCall;
+    updateChatAttributes(request: UpdateChatAttributesRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: UpdateChatAttributesResponse) => void): ClientUnaryCall;
     /** DeleteChats removes chats based on the specified chat_ids */
     deleteChats(request: DeleteChatsRequest, callback: (error: ServiceError | null, response: DeleteChatsResponse) => void): ClientUnaryCall;
     deleteChats(request: DeleteChatsRequest, metadata: Metadata, callback: (error: ServiceError | null, response: DeleteChatsResponse) => void): ClientUnaryCall;
     deleteChats(request: DeleteChatsRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: DeleteChatsResponse) => void): ClientUnaryCall;
+    /** CreateChatAndCompletion makes a new chat when completion present, otherwise sets as new chat */
+    createChatAndCompletion(request: CreateChatAndCompletionRequest, callback: (error: ServiceError | null, response: CreateChatAndCompletionResponse) => void): ClientUnaryCall;
+    createChatAndCompletion(request: CreateChatAndCompletionRequest, metadata: Metadata, callback: (error: ServiceError | null, response: CreateChatAndCompletionResponse) => void): ClientUnaryCall;
+    createChatAndCompletion(request: CreateChatAndCompletionRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: CreateChatAndCompletionResponse) => void): ClientUnaryCall;
+    /** CreateCompletion creates a new completion for an existing chat */
+    createCompletion(request: CreateCompletionRequest, callback: (error: ServiceError | null, response: CreateCompletionResponse) => void): ClientUnaryCall;
+    createCompletion(request: CreateCompletionRequest, metadata: Metadata, callback: (error: ServiceError | null, response: CreateCompletionResponse) => void): ClientUnaryCall;
+    createCompletion(request: CreateCompletionRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: CreateCompletionResponse) => void): ClientUnaryCall;
+    /** DeleteCompletions removes completions based on the specified completion_ids */
+    deleteCompletions(request: DeleteCompletionsRequest, callback: (error: ServiceError | null, response: DeleteCompletionsResponse) => void): ClientUnaryCall;
+    deleteCompletions(request: DeleteCompletionsRequest, metadata: Metadata, callback: (error: ServiceError | null, response: DeleteCompletionsResponse) => void): ClientUnaryCall;
+    deleteCompletions(request: DeleteCompletionsRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: DeleteCompletionsResponse) => void): ClientUnaryCall;
+    /** SearchChatIdsByPromptAndCompletionText searches for completions where the parent chat has the user's ID and either user_prompt_text OR completion_text matches the search_term */
+    searchChatIdsByPromptAndCompletionText(request: SearchChatIdsByPromptAndCompletionTextRequest, callback: (error: ServiceError | null, response: SearchChatIdsByPromptAndCompletionTextResponse) => void): ClientUnaryCall;
+    searchChatIdsByPromptAndCompletionText(request: SearchChatIdsByPromptAndCompletionTextRequest, metadata: Metadata, callback: (error: ServiceError | null, response: SearchChatIdsByPromptAndCompletionTextResponse) => void): ClientUnaryCall;
+    searchChatIdsByPromptAndCompletionText(request: SearchChatIdsByPromptAndCompletionTextRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: SearchChatIdsByPromptAndCompletionTextResponse) => void): ClientUnaryCall;
+    /** UpdateCompletionAttributes updates attribute for a given chat completion */
+    updateCompletionAttributes(request: UpdateCompletionAttributesRequest, callback: (error: ServiceError | null, response: UpdateCompletionAttributesResponse) => void): ClientUnaryCall;
+    updateCompletionAttributes(request: UpdateCompletionAttributesRequest, metadata: Metadata, callback: (error: ServiceError | null, response: UpdateCompletionAttributesResponse) => void): ClientUnaryCall;
+    updateCompletionAttributes(request: UpdateCompletionAttributesRequest, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: UpdateCompletionAttributesResponse) => void): ClientUnaryCall;
+    /** GetCompletionsWithDeepResearcherEntry returns a list of completions that contain metadata on deep researcher */
+    getCompletionsWithDeepResearcherEntry(request: Empty, callback: (error: ServiceError | null, response: GetCompletionsWithDeepResearcherEntryResponse) => void): ClientUnaryCall;
+    getCompletionsWithDeepResearcherEntry(request: Empty, metadata: Metadata, callback: (error: ServiceError | null, response: GetCompletionsWithDeepResearcherEntryResponse) => void): ClientUnaryCall;
+    getCompletionsWithDeepResearcherEntry(request: Empty, metadata: Metadata, options: Partial<CallOptions>, callback: (error: ServiceError | null, response: GetCompletionsWithDeepResearcherEntryResponse) => void): ClientUnaryCall;
 }
 export declare const ApexServiceClient: {
     new (address: string, credentials: ChannelCredentials, options?: Partial<ClientOptions>): ApexServiceClient;