fix: Supported model names

packages/core/src/index.ts

@@ -7,10 +7,10 @@ export type {
 export { executeRequest } from './http-client.js';
 export { getAiCoreDestination } from './context.js';
 export { OpenApiRequestBuilder } from './openapi-request-builder.js';
-export type {
+export {
   AzureOpenAiChatModel,
   AzureOpenAiEmbeddingModel,
   GcpVertexAiChatModel,
   AwsBedrockChatModel,
-  ChatModel
+  AiCoreOpenSourceChatModel
 } from './model-types.js';

packages/core/src/model-types.ts

@@ -9,7 +9,6 @@ export type AzureOpenAiChatModel = LiteralUnion<
   | 'gpt-4'
   | 'gpt-4-32k'
   | 'gpt-35-turbo'
-  | 'gpt-35-turbo-0125'
   | 'gpt-35-turbo-16k'
 >;
 
@@ -24,24 +23,22 @@ export type AzureOpenAiEmbeddingModel = LiteralUnion<
  * GCP Vertex AI models for chat completion.
  */
 export type GcpVertexAiChatModel = LiteralUnion<
-  'gemini-1.0-pro' | 'gemini-1.5-pro' | 'gemini-1.5-flash' | 'chat-bison'
+  'gemini-1.0-pro' | 'gemini-1.5-pro' | 'gemini-1.5-flash'
 >;
 
 /**
  * AWS Bedrock models for chat completion.
  */
 export type AwsBedrockChatModel = LiteralUnion<
-  | 'amazon--titan-text-express'
-  | 'amazon--titan-text-lite'
   | 'anthropic--claude-3-haiku'
   | 'anthropic--claude-3-opus'
   | 'anthropic--claude-3-sonnet'
   | 'anthropic--claude-3.5-sonnet'
 >;
 
 /**
- * All available models for chat completion.
+ * AI Core open source models for chat completion.
  */
-export type ChatModel = LiteralUnion<
-  AzureOpenAiChatModel | GcpVertexAiChatModel | AwsBedrockChatModel
+export type AiCoreOpenSourceChatModel = LiteralUnion<
+  'mistralai--mixtral-8x7b-instruct-v01' | 'meta--llama3-70b-instruct'
 >;

packages/foundation-models/src/azure-openai/azure-openai-chat-client.ts

@@ -1,12 +1,9 @@
-import {
-  type CustomRequestConfig,
-  type AzureOpenAiChatModel,
-  executeRequest
-} from '@sap-ai-sdk/core';
+import { type CustomRequestConfig, executeRequest } from '@sap-ai-sdk/core';
 import {
   getDeploymentId,
   type ModelDeployment
 } from '@sap-ai-sdk/ai-api/internal.js';
+import type { AzureOpenAiChatModel } from './model-types.js';
 import type { AzureOpenAiChatCompletionParameters } from './azure-openai-types.js';
 import { AzureOpenAiChatCompletionResponse } from './azure-openai-response.js';
 

packages/foundation-models/src/azure-openai/azure-openai-embedding-client.ts

@@ -1,12 +1,9 @@
-import {
-  type CustomRequestConfig,
-  type AzureOpenAiEmbeddingModel,
-  executeRequest
-} from '@sap-ai-sdk/core';
+import { type CustomRequestConfig, executeRequest } from '@sap-ai-sdk/core';
 import {
   getDeploymentId,
   type ModelDeployment
 } from '@sap-ai-sdk/ai-api/internal.js';
+import type { AzureOpenAiEmbeddingModel } from './model-types.js';
 import type {
   AzureOpenAiEmbeddingParameters,
   AzureOpenAiEmbeddingOutput

packages/foundation-models/src/azure-openai/index.ts

@@ -2,3 +2,4 @@ export * from './azure-openai-types.js';
 export * from './azure-openai-chat-client.js';
 export * from './azure-openai-embedding-client.js';
 export * from './azure-openai-response.js';
+export * from './model-types.js';

packages/foundation-models/src/azure-openai/model-types.ts

@@ -0,0 +1,14 @@
+import type {
+  AzureOpenAiChatModel as CoreAzureOpenAiChatModel,
+  AzureOpenAiEmbeddingModel as CoreAzureOpenAiEmbeddingModel
+} from '@sap-ai-sdk/core';
+
+/**
+ * Azure OpenAI models for chat completion.
+ */
+export type AzureOpenAiChatModel = CoreAzureOpenAiChatModel;
+
+/**
+ * Azure OpenAI models for embedding.
+ */
+export type AzureOpenAiEmbeddingModel = CoreAzureOpenAiEmbeddingModel;

packages/foundation-models/src/index.ts

@@ -24,8 +24,11 @@ export type {
   AzureOpenAiContentFilterResultBase,
   AzureOpenAiContentFilterDetectedResult,
   AzureOpenAiContentFilterSeverityResult,
-  AzureOpenAiEmbeddingOutput
+  AzureOpenAiEmbeddingOutput,
+  AzureOpenAiChatModel,
+  AzureOpenAiEmbeddingModel
 } from './azure-openai/index.js';
+
 export {
   AzureOpenAiChatClient,
   AzureOpenAiEmbeddingClient,

packages/orchestration/src/index.ts

@@ -39,3 +39,5 @@ export { OrchestrationClient } from './orchestration-client.js';
 export { azureContentFilter } from './orchestration-filter-utility.js';
 
 export { OrchestrationResponse } from './orchestration-response.js';
+
+export type { ChatModel } from './model-types.js';

packages/orchestration/src/model-types.ts

@@ -0,0 +1,15 @@
+import {
+  AiCoreOpenSourceChatModel,
+  AwsBedrockChatModel,
+  AzureOpenAiChatModel,
+  GcpVertexAiChatModel
+} from '@sap-ai-sdk/core';
+
+/**
+ * Supported chat models for orchestration.
+ */
+export type ChatModel =
+  | Exclude<AzureOpenAiChatModel, 'gpt-4o-mini'>
+  | GcpVertexAiChatModel
+  | AwsBedrockChatModel
+  | AiCoreOpenSourceChatModel;

packages/orchestration/src/orchestration-types.ts

@@ -1,4 +1,4 @@
-import { ChatModel } from '@sap-ai-sdk/core';
+import { ChatModel } from './model-types.js';
 import {
   ChatMessages,
   FilteringModuleConfig,

tests/type-tests/test/azure-openai.test-d.ts

@@ -1,5 +1,6 @@
-import { expectError, expectType } from 'tsd';
+import { expectType } from 'tsd';
 import {
+  type AzureOpenAiChatModel,
   AzureOpenAiChatCompletionOutput,
   AzureOpenAiEmbeddingOutput,
   AzureOpenAiChatClient,
@@ -128,3 +129,6 @@ expectType<Promise<AzureOpenAiEmbeddingOutput>>(
     }
   )
 );
+
+expect<AzureOpenAiChatModel>('custom-model');
+expect<AzureOpenAiChatModel>('gpt-4-32k');

tests/type-tests/test/orchestration.test-d.ts

@@ -3,7 +3,8 @@ import {
   OrchestrationClient,
   CompletionPostResponse,
   OrchestrationResponse,
-  TokenUsage
+  TokenUsage,
+  ChatModel
 } from '@sap-ai-sdk/orchestration';
 
 /**
@@ -191,3 +192,6 @@ expectType<Promise<OrchestrationResponse>>(
     }
   }).chatCompletion()
 );
+
+expect<ChatModel>('custom-model');
+expect<ChatModel>('gemini-1.0-pro');