index.js

import { WatsonXAI } from "@ibm-cloud/watsonx-ai";
import wxflows from "@wxflows/sdk/watsonx";
import "dotenv/config";
(async () => {
  process.env.IBM_CREDENTIALS_FILE = "./.env";

  const client = WatsonXAI.newInstance({
    version: "2024-05-31",
    serviceUrl: process.env.WATSONX_AI_ENDPOINT,
  });

  const params = {
    modelId: "mistralai/mistral-large",
    projectId: process.env.WATSONX_AI_PROJECT_ID,
    maxTokens: 100,
  };

  const toolClient = new wxflows({
    endpoint: process.env.WXFLOWS_ENDPOINT,
    apikey: process.env.WXFLOWS_APIKEY,
  });

  const messages = [
    {
      role: "user",
      content: [
        {
          type: "text",
          text: "Search information about the book escape from james patterson",
        },
      ],
    },
  ];

  const tools = await toolClient.tools;

  console.log({ tools })

  const chatCompletion = await client.textChat({
    messages,
    tools,
    ...params,
  });

  const toolMessages = await toolClient.executeTools(chatCompletion);

  console.log({ toolMessages })

  const newMessages = [...messages, ...toolMessages];

  const chatCompleted = await client.textChat({
    messages: newMessages,
    tools,
    ...params,
  });

  console.log(JSON.stringify(chatCompleted));
})();