Skip to content

Commit

Permalink
adjust name
Browse files Browse the repository at this point in the history
  • Loading branch information
deekshas8 committed Jul 11, 2024
1 parent 636cbe9 commit 7d442af
Show file tree
Hide file tree
Showing 4 changed files with 11 additions and 14 deletions.
6 changes: 3 additions & 3 deletions packages/gen-ai-hub/src/client/interface.ts
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import { BaseLlmParameters, CustomRequestConfig } from '../core/http-client.js';
import { GenAiBaseLlmOutput } from './types.js';
import { BaseLlmOutput } from './types.js';

/**
* The base client interface for all provider specific clients.
Expand All @@ -14,7 +14,7 @@ export interface BaseClient<T extends BaseLlmParameters> {
chatCompletion(
data: T,
requestConfig?: CustomRequestConfig
): Promise<GenAiBaseLlmOutput>;
): Promise<BaseLlmOutput>;
/**
* Creates an embedding vector representing the given text.
* @param data - The input parameters for the chat completion.
Expand All @@ -24,5 +24,5 @@ export interface BaseClient<T extends BaseLlmParameters> {
embeddings(
data: T,
requestConfig?: CustomRequestConfig
): Promise<GenAiBaseLlmOutput>;
): Promise<BaseLlmOutput>;
}
2 changes: 1 addition & 1 deletion packages/gen-ai-hub/src/client/types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,6 @@ import {
/**
* Base LLM Output.
*/
export type GenAiBaseLlmOutput =
export type BaseLlmOutput =
| OpenAiChatCompletionOutput
| OpenAiEmbeddingOutput;
3 changes: 1 addition & 2 deletions packages/gen-ai-hub/src/index.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,5 @@ export {
} from './client/index.js';
export { CustomRequestConfig, BaseLlmParameters } from './core/index.js';
export {
getOrchestrationClient,
OrchestrationClientRequest
GenAiHubClient
} from './orchestration/index.js';
14 changes: 6 additions & 8 deletions packages/gen-ai-hub/src/orchestration/orchestration-client.ts
Original file line number Diff line number Diff line change
@@ -1,32 +1,30 @@
import { HttpDestination } from '@sap-cloud-sdk/connectivity';
import { DefaultApi } from './api/default-api.js';
import { CompletionPostRequest } from './api/schema/index.js';
import { CompletionPostRequest, CompletionPostResponse } from './api/schema/index.js';
import { CustomRequestConfig } from '../core/http-client.js';

export type OrchestrationCompletionParameters = Pick<
export type GenAiHubCompletionParameters = Pick<
CompletionPostRequest,
'orchestration_config' | 'return_module_results'
>;

/**
* Get the orchestration client.
*/
export class OrchestrationClient {
export class GenAiHubClient {
destination: HttpDestination;
constructor(destination: HttpDestination) {
this.destination = destination;
}
async chatCompletion(
body: OrchestrationCompletionParameters,
body: GenAiHubCompletionParameters,
requestConfig: CustomRequestConfig
) {
const response = await DefaultApi.orchestrationV1EndpointsCreate({
): Promise<CompletionPostResponse> {
return DefaultApi.orchestrationV1EndpointsCreate({
...body,
input_params: {}
})
.addCustomRequestConfiguration(requestConfig)
.execute(this.destination,);

return response.data;
}
}

0 comments on commit 7d442af

Please sign in to comment.