Skip to main content
LangChain.js supports integration with Gradient AI. Check out Gradient AI for a list of available models.

​
Setup

You’ll need to install the official Gradient Node SDK as a peer dependency:
npm
npm i @gradientai/nodejs-sdk
You will need to set the following environment variables for using the Gradient AI API.
  1. GRADIENT_ACCESS_TOKEN
  2. GRADIENT_WORKSPACE_ID
Alternatively, these can be set during the GradientAI Class instantiation as gradientAccessKey and workspaceId respectively. For example: 
const model = new GradientLLM({
  gradientAccessKey: "My secret Access Token"
  workspaceId: "My secret workspace id"
});

​
Usage

See this section for general instructions on installing LangChain packages.
npm
npm install @langchain/community @langchain/core

import { GradientLLM } from "@langchain/community/llms/gradient_ai";

// Note that inferenceParameters are optional
const model = new GradientLLM({
  modelSlug: "llama2-7b-chat",
  inferenceParameters: {
    maxGeneratedTokenCount: 20,
    temperature: 0,
  },
});
const res = await model.invoke(
  "What would be a good company name for a company that makes colorful socks?"
);

console.log({ res });

​
Using your own fine-tuned Adapters

The use your own custom adapter simply set adapterId during setup.
Edit the source of this page on GitHub.
Connect these docs programmatically to Claude, VSCode, and more via MCP for real-time answers.