feat: add reasoning model support 🤖

microsoft · Dec 17, 2024 · 1cc5884 · 1cc5884
1 parent c6a9828
commit 1cc5884
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/packages/core/src/types/prompt_template.d.ts b/packages/core/src/types/prompt_template.d.ts
@@ -122,6 +122,8 @@ type ModelType = OptionsOrString<
     | "large"
     | "small"
     | "vision"
+    | "reasoning"
+    | "reasoning_small"
     | "openai:gpt-4o"
     | "openai:gpt-4o-mini"
     | "openai:gpt-3.5-turbo"
@@ -219,7 +221,9 @@ interface ModelOptions extends ModelConnectionOptions, ModelTemplateOptions {
     /**
      * A list of keywords that should be found in the output.
      */
-    choices?: ElementOrArray<string | { token: string | number; weight?: number }>
+    choices?: ElementOrArray<
+        string | { token: string | number; weight?: number }
+    >
 
     /**
      * Returns the log probabilities of the each tokens. Not supported in all models.

diff --git a/packages/sample/genaisrc/samples/gai.genai.mts b/packages/sample/genaisrc/samples/gai.genai.mts
@@ -3,6 +3,7 @@
 // Script for analyzing GitHub Action runs to determine the cause of a failure.
 script({
     title: "GitHub Action Investigator",
+    model: "reasoning",
     description:
         "Analyze GitHub Action runs to find the root cause of a failure",
     parameters: {