@ant-design/x/es/use-x-agent/index.js

Craft AI-driven interfaces effortlessly

import React from 'react';
import XRequest from "../x-request";
let uuid = 0;

/** This is a wrap class to avoid developer can get too much on origin object */
export class XAgent {
  config;
  requestingMap = {};
  constructor(config) {
    this.config = config;
  }
  finishRequest(id) {
    delete this.requestingMap[id];
  }
  request = (info, callbacks) => {
    const {
      request
    } = this.config;
    const {
      onUpdate,
      onSuccess,
      onError
    } = callbacks;
    const id = uuid;
    uuid += 1;
    this.requestingMap[id] = true;
    request?.(info, {
      // Status should be unique.
      // One get success or error should not get more message
      onUpdate: message => {
        if (this.requestingMap[id]) {
          onUpdate(message);
        }
      },
      onSuccess: message => {
        if (this.requestingMap[id]) {
          onSuccess(message);
          this.finishRequest(id);
        }
      },
      onError: error => {
        if (this.requestingMap[id]) {
          onError(error);
          this.finishRequest(id);
        }
      }
    });
  };
  isRequesting() {
    return Object.keys(this.requestingMap).length > 0;
  }
}
export default function useXAgent(config) {
  const {
    request,
    ...restConfig
  } = config;
  return React.useMemo(() => [new XAgent({
    request: request || XRequest({
      baseURL: restConfig.baseURL,
      model: restConfig.model,
      dangerouslyApiKey: restConfig.dangerouslyApiKey
    }).create,
    ...restConfig
  })], []);
}