notes

Author: Hendrixer

README.md

@@ -0,0 +1,5 @@
+# AI with Node.js
+
+## Materials
+
+- [notes](https://scottmoss.notion.site/AI-App-Node-js-f9a372a138ef4241943b4fbb44bdc970?pvs=4)

functions.js

@@ -0,0 +1,77 @@
+import 'dotenv/config'
+import { openai } from './openai.js'
+import math from 'advanced-calculator'
+const QUESTION = process.argv[2] || 'hi'
+
+const messages = [
+  {
+    role: 'user',
+    content: QUESTION,
+  },
+]
+
+const functions = {
+  calculate: async ({ expression }) => {
+    return math.evaluate(expression)
+  },
+}
+
+const getCompletion = async (messages) => {
+  const response = await openai.chat.completions.create({
+    model: 'gpt-3.5-turbo-0613',
+    messages,
+    functions: [
+      {
+        name: 'calculate',
+        description: 'Run a math expression',
+        parameters: {
+          type: 'object',
+          properties: {
+            expression: {
+              type: 'string',
+              description:
+                'Then math expression to evaluate like "2 * 3 + (21 / 2) ^ 2"',
+            },
+          },
+          required: ['expression'],
+        },
+      },
+    ],
+    temperature: 0,
+  })
+
+  return response
+}
+
+let response
+while (true) {
+  response = await getCompletion(messages)
+
+  if (response.choices[0].finish_reason === 'stop') {
+    console.log(response.choices[0].message.content)
+    break
+  } else if (response.choices[0].finish_reason === 'function_call') {
+    const fnName = response.choices[0].message.function_call.name
+    const args = response.choices[0].message.function_call.arguments
+
+    const functionToCall = functions[fnName]
+    const params = JSON.parse(args)
+
+    const result = functionToCall(params)
+
+    messages.push({
+      role: 'assistant',
+      content: null,
+      function_call: {
+        name: fnName,
+        arguments: args,
+      },
+    })
+
+    messages.push({
+      role: 'function',
+      name: fnName,
+      content: JSON.stringify({ result: result }),
+    })
+  }
+}

index.js

@@ -11,6 +11,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "advanced-calculator": "^1.1.1",
     "dotenv": "^16.3.1",
     "langchain": "^0.0.170",
     "openai": "^4.12.4",

package-lock.json

@@ -47,7 +47,7 @@ const loadStore = async () => {
 
 const query = async () => {
   const store = await loadStore()
-  const results = await store.similaritySearch(question, 2)
+  const results = await store.similaritySearch(question, 1)
 
   const response = await openai.chat.completions.create({
     model: 'gpt-3.5-turbo-16k-0613',