call endpoint via a langchain chatapp

examples/rag-question-answer/app/package.json

@@ -10,13 +10,15 @@
     "@carbon/react": "^1.37.0",
     "@graphql-tools/utils": "^10.1.3",
     "@vitejs/plugin-react": "^4.0.4",
+    "chatwxflows": "file:./chatwxflows-0.1.0.tgz",
     "graphql": "^16.8.1",
     "graphql-request": "^6.1.0",
     "react": "^18.2.0",
     "react-dom": "^18.2.0",
     "vite": "^4.4.9",
     "watzen": "^0.1.10",
-    "wxflows": "^0.1.0"
+    "wxflows": "^0.1.1",
+    "zod": "^3.24.1"
   },
   "devDependencies": {
     "sass": "^1.66.1"

examples/rag-question-answer/app/src/App.jsx

@@ -22,7 +22,7 @@ function App() {
         throw new Error('Something went wrong')
       }
     } catch (e) {
-      console.error('Something went wrong')
+      console.error('Something went wrong', e)
     } finally {
       setIsLoading(false)
     }

examples/rag-question-answer/app/src/callWxflows.js

@@ -1,34 +1,29 @@
-import wxflows from 'wxflows'
+import { ChatWXFlows } from 'chatwxflows'
+import { calculatorTool } from './tools';
 
 export async function getAnswer(question) {
-    const model = new wxflows({
+    const chatModel = new ChatWXFlows({ 
+        n: 4, 
         endpoint: import.meta.env.VITE_WXFLOWS_ENDPOINT,
         apikey: import.meta.env.VITE_WXFLOWS_APIKEY,
-    })
-
-    const schema = await model.generate()
-
-    // Make sure these match your values in `wxflows.toml`
-    const flowName = 'myRag'
-    const collection = import.meta.env.VITE_WXFLOWS_COLLECTION
-
-    const result = await model.ragAnswer({
-        schema,
-        flowName,
+        flowName: 'myRag',
         variables: {
             n: 5,
-            question,
             aiEngine: 'WATSONX',
             model: 'ibm/granite-13b-chat-v2',
-            collection,
+            collection: 'watsonxdocs',
             parameters: {
                 max_new_tokens: 1000,
                 temperature: 0.7,
                 stop_sequences: ['\n\n'],
             },
             searchEngine: 'GETTINGSTARTED',
-        },
-    })
+        }
+    });
+
+    const llmWithTools = chatModel.bindTools([calculatorTool]);
 
-    return result?.data?.[flowName]?.out?.modelResponse?.results[0]?.generated_text;
-}
+    const res = await llmWithTools.invoke(question);
+
+    return res.content
+}

examples/rag-question-answer/app/src/tools.tsx

@@ -0,0 +1,49 @@
+import { tool } from "@langchain/core/tools";
+import { z } from "zod";
+
+export const calculatorSchema = z.object({
+  operation: z
+    .enum(["add", "subtract", "multiply", "divide"])
+    .describe("The type of operation to execute."),
+  number1: z.number().describe("The first number to operate on."),
+  number2: z.number().describe("The second number to operate on."),
+});
+
+export const calculatorTool = tool(async ({ operation, number1, number2 }) => {
+    if (operation === "add") {
+      return `${number1 + number2}`;
+    } else if (operation === "subtract") {
+      return `${number1 - number2}`;
+    } else if (operation === "multiply") {
+      return `${number1 * number2}`;
+    } else if (operation === "divide") {
+      return `${number1 / number2}`;
+    } else {
+      throw new Error("Invalid operation.");
+    }
+  }, {
+    name: "calculator",
+    description: "Can perform mathematical operations.",
+    schema: calculatorSchema,
+  });
+
+
+export const getCalculatorToolML = async (calculatorTool: any, userMessage: string, operation: string) => {
+    // Extract operation and numbers using a regex
+    const match = userMessage.match(/What is (\d+) \* (\d+)/);
+    if (!match) {
+      throw new Error("Could not parse the question.");
+    }
+
+    const number1 = parseFloat(match[1]);
+    const number2 = parseFloat(match[2]);
+
+    // Invoke the calculator tool
+    const result = await calculatorTool.invoke({
+      operation,
+      number1,
+      number2,
+    });
+
+    return `The result of ${number1} * ${number2} is ${result}.`;
+}