mirror of
https://github.com/danny-avila/LibreChat.git
synced 2025-09-22 08:12:00 +02:00
56ea0f9ae7
* refactor(deploy-compose.yml): use long-syntax to avoid implicit folder creation of librechat.yaml * refactor(docker-compose.override.yml.example): use long-syntax to avoid implicit folder creation of librechat.yaml * chore: add simple health check for RAG_API_URL * chore: improve axios error handling, adding `logAxiosError` * chore: more informative message detailing RAG_API_URL path * feat: add rag_api and vectordb to default compose file * chore(rag.yml): update standalone rag compose file to use RAG_PORT * chore: documentation updates * docs: Update rag_api.md with images * Update rag_api.md * Update rag_api.md, assistants clarification * add RAG API note to breaking changes
63 lines
2 KiB
JavaScript
63 lines
2 KiB
JavaScript
const axios = require('axios');
|
|
const { EModelEndpoint } = require('librechat-data-provider');
|
|
const { logAxiosError } = require('~/utils');
|
|
|
|
/**
|
|
* @typedef {Object} RetrieveOptions
|
|
* @property {string} thread_id - The ID of the thread to retrieve.
|
|
* @property {string} run_id - The ID of the run to retrieve.
|
|
* @property {number} [timeout] - Optional timeout for the API call.
|
|
* @property {number} [maxRetries] - TODO: not yet implemented; Optional maximum number of retries for the API call.
|
|
* @property {OpenAIClient} openai - Configuration and credentials for OpenAI API access.
|
|
*/
|
|
|
|
/**
|
|
* Asynchronously retrieves data from an API endpoint based on provided thread and run IDs.
|
|
*
|
|
* @param {RetrieveOptions} options - The options for the retrieve operation.
|
|
* @returns {Promise<Object>} The data retrieved from the API.
|
|
*/
|
|
async function retrieveRun({ thread_id, run_id, timeout, openai }) {
|
|
const { apiKey, baseURL, httpAgent, organization } = openai;
|
|
let url = `${baseURL}/threads/${thread_id}/runs/${run_id}`;
|
|
|
|
let headers = {
|
|
Authorization: `Bearer ${apiKey}`,
|
|
'OpenAI-Beta': 'assistants=v1',
|
|
};
|
|
|
|
if (organization) {
|
|
headers['OpenAI-Organization'] = organization;
|
|
}
|
|
|
|
/** @type {TAzureConfig | undefined} */
|
|
const azureConfig = openai.req.app.locals[EModelEndpoint.azureOpenAI];
|
|
|
|
if (azureConfig && azureConfig.assistants) {
|
|
delete headers.Authorization;
|
|
headers = { ...headers, ...openai._options.defaultHeaders };
|
|
const queryParams = new URLSearchParams(openai._options.defaultQuery).toString();
|
|
url = `${url}?${queryParams}`;
|
|
}
|
|
|
|
try {
|
|
const axiosConfig = {
|
|
headers: headers,
|
|
timeout: timeout,
|
|
};
|
|
|
|
if (httpAgent) {
|
|
axiosConfig.httpAgent = httpAgent;
|
|
axiosConfig.httpsAgent = httpAgent;
|
|
}
|
|
|
|
const response = await axios.get(url, axiosConfig);
|
|
return response.data;
|
|
} catch (error) {
|
|
const message = '[retrieveRun] Failed to retrieve run data:';
|
|
logAxiosError({ message, error });
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
module.exports = { retrieveRun };
|