Browse Source

Fix o1 on openai native

Saoud Rizwan 1 year ago
parent
commit
dff81b9d09
2 changed files with 13 additions and 3 deletions
  1. 2 2
      package-lock.json
  2. 11 1
      src/api/providers/openai-native.ts

+ 2 - 2
package-lock.json

@@ -1,12 +1,12 @@
 {
   "name": "claude-dev",
-  "version": "2.0.5",
+  "version": "2.0.7",
   "lockfileVersion": 3,
   "requires": true,
   "packages": {
     "": {
       "name": "claude-dev",
-      "version": "2.0.5",
+      "version": "2.0.7",
       "license": "Apache-2.0",
       "dependencies": {
         "@anthropic-ai/bedrock-sdk": "^0.10.2",

+ 11 - 1
src/api/providers/openai-native.ts

@@ -23,8 +23,18 @@ export class OpenAiNativeHandler implements ApiHandler {
 	}
 
 	async *createMessage(systemPrompt: string, messages: Anthropic.Messages.MessageParam[]): ApiStream {
+		let systemPromptMessage: OpenAI.Chat.ChatCompletionMessageParam
+		switch (this.getModel().id) {
+			case "o1-preview":
+			case "o1-mini":
+				systemPromptMessage = { role: "user", content: systemPrompt }
+				break
+			default:
+				systemPromptMessage = { role: "system", content: systemPrompt }
+		}
+
 		const openAiMessages: OpenAI.Chat.ChatCompletionMessageParam[] = [
-			{ role: "system", content: systemPrompt },
+			systemPromptMessage,
 			...convertToOpenAiMessages(messages),
 		]