Merge pull request cloudflare#146 from rcccf/rchen/fix-ai

Fix OpenAPI endpoint usage
code · Dec 20, 2024 · c143f77 · c143f77
2 parents 778f5f6 + c80dc5e
commit c143f77
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/app/durableObjects/ChatRoom.server.ts b/app/durableObjects/ChatRoom.server.ts
@@ -25,6 +25,7 @@ import {
 } from '~/utils/openai.server'
 
 const alarmInterval = 15_000
+const defaultOpenAIModelID = 'gpt-4o-realtime-preview-2024-10-01'
 
 /**
  * The ChatRoom Durable Object Class
@@ -415,6 +416,8 @@ export class ChatRoom extends Server<Env> {
 							params.set('instructions', instructions)
 						}
 
+						params.set('model', this.env.OPENAI_MODEL_ID || defaultOpenAIModelID)
+
 						// The Calls's offer is sent to OpenAI
 						const openaiAnswer = await requestOpenAIService(
 							openAiTracksResponse.sessionDescription ||

diff --git a/app/types/Env.ts b/app/types/Env.ts
@@ -17,4 +17,5 @@ export type Env = {
 	DB?: D1Database
 	OPENAI_API_TOKEN?: string
 	OPENAI_MODEL_ENDPOINT?: string
+	OPENAI_MODEL_ID?: string
 }