mirror of
https://github.com/hanzoai/chat.git
synced 2026-09-30 20:51:17 +00:00
zooqueen
48eac75e13
Forks the internal @librechat/api workspace package (packages/api) to @hanzochat/api, matching the app package @hanzochat/chat. De-LibreChats the package name and puts its build under our control. Fixes the live prod bug where the deployed chat image shipped a stale @librechat/api/dist missing the exported handleError(), breaking every error path with "handleError is not a function" and killing the SSE stream. A clean build of @hanzochat/api produces dist/index.js that exports handleError (verified: exports.handleError = handleError). - packages/api/package.json: name @librechat/api -> @hanzochat/api - 254 source/config/doc/lockfile sites: @librechat/api -> @hanzochat/api (api/, packages/, config/, turbo.json, pnpm-lock.yaml, bun.lock, package-lock.json, LLM.md) - @librechat/api-client (distinct MCP client id string) left untouched - other @librechat/* packages (client, data-schemas, data-provider) left as-is per scope pnpm install (regenerated lockfile) + build:api green; dist exports handleError. api/data-schemas/client packages build; full build:packages and api test:ci are blocked by a PRE-EXISTING, unrelated upstream break in librechat-data-provider (config.spec.ts / cloudfront-config.spec.ts reference exports that src/config.ts no longer has) -- out of scope here.
127 lines
3.4 KiB
JavaScript
127 lines
3.4 KiB
JavaScript
const { getModelMaxTokens } = require('@hanzochat/api');
|
|
const BaseClient = require('../BaseClient');
|
|
|
|
class FakeClient extends BaseClient {
|
|
constructor(apiKey, options = {}) {
|
|
super(apiKey, options);
|
|
this.sender = 'AI Assistant';
|
|
this.setOptions(options);
|
|
}
|
|
setOptions(options) {
|
|
if (this.options && !this.options.replaceOptions) {
|
|
this.options.modelOptions = {
|
|
...this.options.modelOptions,
|
|
...options.modelOptions,
|
|
};
|
|
delete options.modelOptions;
|
|
this.options = {
|
|
...this.options,
|
|
...options,
|
|
};
|
|
} else {
|
|
this.options = options;
|
|
}
|
|
|
|
if (this.options.openaiApiKey) {
|
|
this.apiKey = this.options.openaiApiKey;
|
|
}
|
|
|
|
const modelOptions = this.options.modelOptions || {};
|
|
if (!this.modelOptions) {
|
|
this.modelOptions = {
|
|
...modelOptions,
|
|
model: modelOptions.model || 'gpt-3.5-turbo',
|
|
temperature:
|
|
typeof modelOptions.temperature === 'undefined' ? 0.8 : modelOptions.temperature,
|
|
top_p: typeof modelOptions.top_p === 'undefined' ? 1 : modelOptions.top_p,
|
|
presence_penalty:
|
|
typeof modelOptions.presence_penalty === 'undefined' ? 1 : modelOptions.presence_penalty,
|
|
stop: modelOptions.stop,
|
|
};
|
|
}
|
|
|
|
this.maxContextTokens =
|
|
this.options.maxContextTokens ?? getModelMaxTokens(this.modelOptions.model) ?? 4097;
|
|
}
|
|
buildMessages() {}
|
|
getTokenCount(str) {
|
|
return str.length;
|
|
}
|
|
getTokenCountForMessage(message) {
|
|
return message?.content?.length || message.length;
|
|
}
|
|
}
|
|
|
|
const initializeFakeClient = (apiKey, options, fakeMessages) => {
|
|
let TestClient = new FakeClient(apiKey);
|
|
TestClient.options = options;
|
|
TestClient.abortController = { abort: jest.fn() };
|
|
TestClient.loadHistory = jest
|
|
.fn()
|
|
.mockImplementation((conversationId, parentMessageId = null) => {
|
|
if (!conversationId) {
|
|
TestClient.currentMessages = [];
|
|
return Promise.resolve([]);
|
|
}
|
|
|
|
const orderedMessages = TestClient.constructor.getMessagesForConversation({
|
|
messages: fakeMessages,
|
|
parentMessageId,
|
|
});
|
|
|
|
TestClient.currentMessages = orderedMessages;
|
|
return Promise.resolve(orderedMessages);
|
|
});
|
|
|
|
TestClient.getSaveOptions = jest.fn().mockImplementation(() => {
|
|
return {};
|
|
});
|
|
|
|
TestClient.getBuildMessagesOptions = jest.fn().mockImplementation(() => {
|
|
return {};
|
|
});
|
|
|
|
TestClient.sendCompletion = jest.fn(async () => {
|
|
return {
|
|
completion: 'Mock response text',
|
|
metadata: undefined,
|
|
};
|
|
});
|
|
|
|
TestClient.getCompletion = jest.fn().mockImplementation(async (..._args) => {
|
|
return {
|
|
choices: [
|
|
{
|
|
message: {
|
|
content: 'Mock response text',
|
|
},
|
|
},
|
|
],
|
|
};
|
|
});
|
|
|
|
TestClient.buildMessages = jest.fn(async (messages, parentMessageId) => {
|
|
const orderedMessages = TestClient.constructor.getMessagesForConversation({
|
|
messages,
|
|
parentMessageId,
|
|
});
|
|
const formattedMessages = orderedMessages.map((message) => {
|
|
let { role: _role, sender, text } = message;
|
|
const role = _role ?? sender;
|
|
const content = text ?? '';
|
|
return {
|
|
role: role?.toLowerCase() === 'user' ? 'user' : 'assistant',
|
|
content,
|
|
};
|
|
});
|
|
return {
|
|
prompt: formattedMessages,
|
|
tokenCountMap: null, // Simplified for the mock
|
|
};
|
|
});
|
|
|
|
return TestClient;
|
|
};
|
|
|
|
module.exports = { FakeClient, initializeFakeClient };
|