From dff81b9d09455024727da046946128acc8867ceb Mon Sep 17 00:00:00 2001
From: Saoud Rizwan <7799382+saoudrizwan@users.noreply.github.com>
Date: Wed, 16 Oct 2024 09:59:30 -0400
Subject: [PATCH] Fix o1 on openai native

---
 package-lock.json                  |  4 ++--
 src/api/providers/openai-native.ts | 12 +++++++++++-
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/package-lock.json b/package-lock.json
index 44ad675..146a0c9 100644
--- a/package-lock.json
+++ b/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",
diff --git a/src/api/providers/openai-native.ts b/src/api/providers/openai-native.ts
index 2245553..f21f036 100644
--- a/src/api/providers/openai-native.ts
+++ b/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),
 		]