update: release v1.7.6

elsoul · Nov 7, 2023 · 0bf43f2 · 0bf43f2
1 parent 4906b3d
commit 0bf43f2
Show file tree

Hide file tree

Showing 31 changed files with 66 additions and 64 deletions.
diff --git a/dist/lib/genPrompt.d.ts b/dist/lib/genPrompt.d.ts
@@ -1,4 +1,5 @@
-import { OpenAIPromptParams, VertexPromptParams } from './types';
+import { VertexPromptParams } from './types';
+import { ChatCompletionMessageParam } from 'openai/resources';
 /**
  * Represents the AI platforms supported by the generatePrompt function.
  */
@@ -29,7 +30,9 @@ export interface AIPrompt {
      */
     examples: AIExample[];
 }
-export declare function generatePrompt(context: string, examples: AIExample[], content: string, ai: AIType): VertexPromptParams | OpenAIPromptParams;
+export declare function generatePrompt(context: string, examples: AIExample[], content: string, ai: AIType): VertexPromptParams | {
+    messages: ChatCompletionMessageParam[];
+};
 export declare const migrationPrompt: {
     context: string;
     examples: {

diff --git a/dist/lib/version.d.ts b/dist/lib/version.d.ts
@@ -1 +1 @@
-export declare const VERSION = "1.7.5";
+export declare const VERSION = "1.7.6";
diff --git a/docs/classes/OpenAI-1.html b/docs/classes/OpenAI-1.html
diff --git a/docs/classes/SkeetAI.html b/docs/classes/SkeetAI.html
diff --git a/docs/classes/VertexAI.html b/docs/classes/VertexAI.html
diff --git a/docs/enums/InstanceType.html b/docs/enums/InstanceType.html
diff --git a/docs/enums/NamingEnum.html b/docs/enums/NamingEnum.html
diff --git a/docs/enums/SkeetAiMode.html b/docs/enums/SkeetAiMode.html
diff --git a/docs/functions/generatePrompt.html b/docs/functions/generatePrompt.html
diff --git a/docs/functions/translate.html b/docs/functions/translate.html
diff --git a/docs/interfaces/AIPrompt.html b/docs/interfaces/AIPrompt.html
diff --git a/docs/interfaces/SkeetAIOptions.html b/docs/interfaces/SkeetAIOptions.html
diff --git a/docs/types/AIExample.html b/docs/types/AIExample.html
diff --git a/docs/types/AIType.html b/docs/types/AIType.html
diff --git a/docs/types/AiInstance.html b/docs/types/AiInstance.html
diff --git a/docs/types/Example.html b/docs/types/Example.html
diff --git a/docs/types/InputOutput.html b/docs/types/InputOutput.html
diff --git a/docs/types/OpenAIMessage.html b/docs/types/OpenAIMessage.html
diff --git a/docs/types/OpenAIOptions.html b/docs/types/OpenAIOptions.html
diff --git a/docs/types/OpenAIPromptParams.html b/docs/types/OpenAIPromptParams.html
diff --git a/docs/types/OpenAIRole.html b/docs/types/OpenAIRole.html
diff --git a/docs/types/VertexAiOptions.html b/docs/types/VertexAiOptions.html
diff --git a/docs/types/VertexExample.html b/docs/types/VertexExample.html
diff --git a/docs/types/VertexExampleMessage.html b/docs/types/VertexExampleMessage.html
diff --git a/docs/types/VertexMessage.html b/docs/types/VertexMessage.html
diff --git a/docs/types/VertexParameterParams.html b/docs/types/VertexParameterParams.html
diff --git a/docs/types/VertexPromptParams.html b/docs/types/VertexPromptParams.html
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@skeet-framework/ai",
-  "version": "1.7.5",
+  "version": "1.7.6",
   "description": "Skeet Framework Plugin - AI",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

diff --git a/src/lib/genPrompt.ts b/src/lib/genPrompt.ts
@@ -1,5 +1,6 @@
 import { inspect } from 'util'
 import { OpenAIPromptParams, VertexExample, VertexPromptParams } from './types'
+import { ChatCompletionMessageParam } from 'openai/resources'
 
 /**
  * Represents the AI platforms supported by the generatePrompt function.
@@ -41,7 +42,7 @@ export function generatePrompt(
   examples: AIExample[],
   content: string,
   ai: AIType,
-): VertexPromptParams | OpenAIPromptParams {
+): VertexPromptParams | { messages: ChatCompletionMessageParam[] } {
   if (ai === 'VertexAI') {
     const exampleMessages = []
     for (const example of examples) {
@@ -71,14 +72,14 @@ export function generatePrompt(
         exampleMessages.push({
           role: 'user',
           content: example.input,
-        })
+        } as ChatCompletionMessageParam)
       if (!example.output) continue
       exampleMessages.push({
         role: 'assistant',
         content: example.output,
-      })
+      } as ChatCompletionMessageParam)
     }
-    const messages = [
+    const messages: ChatCompletionMessageParam[] = [
       {
         role: 'system',
         content: context,
@@ -90,9 +91,7 @@ export function generatePrompt(
       },
     ]
 
-    return {
-      messages,
-    } as OpenAIPromptParams
+    return { messages }
   } else {
     throw new Error('Unsupported AI type')
   }
@@ -156,4 +155,4 @@ export const namingPrompt = {
       output: 'fetchLatestNews',
     },
   ],
-}
+}
diff --git a/src/lib/openai/exapmle.ts b/src/lib/openai/exapmle.ts
@@ -8,25 +8,25 @@ const exampleJosn = JSON.parse(
 )
 
 const run = async () => {
-  // const content = 'What is the skeet framework?'
-  // const prompt = generatePrompt(
-  //   exampleJosn.context,
-  //   exampleJosn.examples,
-  //   content,
-  //   'OpenAI',
-  // ) as OpenAIPromptParams
+  const content = 'What is the skeet framework?'
+  const prompt = generatePrompt(
+    exampleJosn.context,
+    exampleJosn.examples,
+    content,
+    'OpenAI',
+  )
   const openAi = new SkeetAI({
     ai: 'OpenAI',
     model: 'gpt-4-1106-preview',
   }).aiInstance as OpenAI
-  // const result = await openAi.prompt(prompt)
+  const result = await openAi.prompt(prompt)
 
-  // console.log('Question:\n', prompt.messages[3].content)
-  // console.log('\nAnswer:\n', result)
-  const content2 =
-    'The Skeet framework is an open-source full-stack app development solution that aims to lower the development and operation cost of applications. It allows developers to focus more on the application logic and worry less about infrastructure. The framework can be assembled with a combination of SQL and NoSQL.'
-  const title = await openAi.generateTitle(content2)
-  console.log('\nGenerated title:\n', title)
+  console.log('Question:\n', content)
+  console.log('\nAnswer:\n', result)
+  // const content2 =
+  //   'The Skeet framework is an open-source full-stack app development solution that aims to lower the development and operation cost of applications. It allows developers to focus more on the application logic and worry less about infrastructure. The framework can be assembled with a combination of SQL and NoSQL.'
+  // const title = await openAi.generateTitle(content2)
+  // console.log('\nGenerated title:\n', title)
 }
 
 run()
diff --git a/src/lib/version.ts b/src/lib/version.ts
@@ -1 +1 @@
-export const VERSION = '1.7.5'
+export const VERSION = '1.7.6'
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		export declare const VERSION = "1.7.5";
		export declare const VERSION = "1.7.6";
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		export const VERSION = '1.7.5'
		export const VERSION = '1.7.6'