added o1 support

Author: carusyte

.vscode/extensions.json

@@ -1,5 +1,9 @@
 {
   // See http://go.microsoft.com/fwlink/?LinkId=827846
   // for the documentation about the extensions.json format
-  "recommendations": ["dbaeumer.vscode-eslint", "amodio.tsl-problem-matcher"]
+  "recommendations": [
+    "dbaeumer.vscode-eslint",
+    "amodio.tsl-problem-matcher",
+    "carusyte.gai-choy"
+  ]
 }

CHANGELOG.md

@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.4.0] - 2025-03-10
+
+- Added support for o1 model.
+
 ## [0.3.0] - 2024-11-13
 
 - Added support for o1-preview and o1-mini models.

package.json

@@ -3,7 +3,7 @@
   "displayName": "GAI Choy",
   "description": "G̲enerative A̲I̲ empowered, C̲ode H̲elper O̲n Y̲our side. Yet another Copilot for coding, with built-in integration capability with Azure OpenAI models or, local LLM such as CodeShell.",
   "publisher": "carusyte",
-  "version": "0.3.0",
+  "version": "0.4.0",
   "icon": "assets/logo.png",
   "pricing": "Free",
   "keywords": [
@@ -64,7 +64,8 @@
             "gpt-4-32k",
             "gpt-4o",
             "o1-mini",
-            "o1-preview"
+            "o1-preview",
+            "o1"
           ],
           "type": "string",
           "order": 4
@@ -79,7 +80,8 @@
             "gpt-4-32k",
             "gpt-4o",
             "o1-mini",
-            "o1-preview"
+            "o1-preview",
+            "o1"
           ],
           "type": "string",
           "order": 5
@@ -110,7 +112,9 @@
             "2024-02-01",
             "2024-05-01-preview",
             "2024-08-01-preview",
-            "2024-10-21"
+            "2024-10-21",
+            "2024-12-01-preview",
+            "2025-01-01-preview"
           ],
           "type": "string",
           "order": 8

src/RequestEventStream.ts

@@ -69,7 +69,7 @@ export async function postEventStream(prompt: string, chatList: Array<ChatItem>,
             temperature: 0.8,
             messages: [
                 {
-                    role: model === "o1-preview" ? "user" : "system",
+                    role: (/^o1(-.*)?$/.test(model)) ? "user" : "system",
                     // "content": `
                     //     Your role is an AI pair programming assistant and technical consultant, a programming expert with strong coding skills.
                     //     Your task is to answer questions raised by the user as a developer.
@@ -95,7 +95,7 @@ export async function postEventStream(prompt: string, chatList: Array<ChatItem>,
 
         AzureOAI.mergeParameters(body, parameters)
 
-        if (model === "o1-preview") {
+        if (/^o1(-.*)?$/.test(model)) {
            delete body.temperature // this parameter is not supported by o1 model
         }
 

src/llm/AzureOAI.ts

@@ -50,7 +50,7 @@ export class AzureOAI {
             "temperature": 0.2,
             "messages": [
                 {
-                    "role": model === "o1-preview" ? "user": "system",
+                    "role": (/^o1(-.*)?$/.test(model)) ? "user": "system",
                     "content": `Your role is an AI code generator. Your name is "GAI Choy".
 Your task is to provide executable and functional code fragments AS-IS, based on the context provided by the user.
 The context and metadata of the code fragment will be provided by user in the following format, as surrounded by triple-backticks.
@@ -120,7 +120,7 @@ Expected response in JSON format:
             ...data, response_format: { type: 'json_object' }
         } : data
 
-        if (model === "o1-preview") {
+        if (/^o1(-.*)?$/.test(model)) {
             delete data.temperature // this parameter is not supported by o1 model
         }