Knowledge base

To use a private, secure and encrypted Knowledge base:

Create a Knowledge base.

  1. Login to PropulsionAI

  2. Goto Knowledge base.

  3. Create a new Knowledge base.

  4. Upload documents to Knowledge base.

circle-info

For large documents, wait for document processing to be completed.

Inference

import PropulsionAI from 'propulsionai';
import {
  ModelChatResponse,
  ModelChatParams
} from 'propulsionai/resources/models';

const p8n = new PropulsionAI({
  bearerToken: process.env['PROPULSIONAI_BEARER_TOKEN'] as string,
});

async function main() {
  try {
    let deployment_tag: string = '<deployment_tag>';
    let knowledgebases: Array<string> = ["<knowledgebase_id>"]
    let params: ModelChatParams = {
      knowledgebases: knowledgebases,
      messages: [
        {
          role: 'user',
          content: "What are the key improvements in US-X2090249-B00?",
        },
      ],
      model: 'auto',
      stream: false,
      wait: true,
    };
    let response: ModelChatResponse;
    try{
      response = await p8n.models.ep(deployment_tag, params);
    }catch(e){
      console.log('Error:', e);
      return;
    }
    console.log("Task ID:", response.task_id);
    if(response.choices) {
      response.choices.forEach((choice) => {
        console.log(`${choice.message?.role}: ${choice.message?.content}`);
      });
    }
  } catch (error) {
    console.error('Error during model call:', error);
  }
}

main();

Last updated