"feat: 增加助手应用配置管理功能与服务器守护进程支持"
This commit is contained in:
53
assistant/src/test/chat.ts
Normal file
53
assistant/src/test/chat.ts
Normal file
@@ -0,0 +1,53 @@
|
||||
// import { AssistantConfig } from '@/module/assistant/index.ts';
|
||||
import { assistantConfig } from '../config.ts';
|
||||
|
||||
import { ProviderManager, SiliconFlowProvider } from '@kevisual/ai-center';
|
||||
|
||||
const config = assistantConfig.getConfig();
|
||||
console.log('aiConfig', config.ai);
|
||||
const providerManager = new ProviderManager({
|
||||
provider: config.ai.provider,
|
||||
apiKey: config.ai.apiKey!,
|
||||
model: config.ai.model!,
|
||||
});
|
||||
|
||||
const chatTest = async (text: string) => {
|
||||
const result = await providerManager.provider.chat([
|
||||
{
|
||||
role: 'user',
|
||||
content: text,
|
||||
},
|
||||
]);
|
||||
console.log('result', result);
|
||||
return result.choices[0].message.content;
|
||||
};
|
||||
|
||||
chatTest('你好').then((res) => {
|
||||
console.log('chatTest', res);
|
||||
});
|
||||
|
||||
// providerManager.provider.test().then((res) => {
|
||||
// console.log('test', res);
|
||||
// });
|
||||
// const siliconflow = providerManager.provider as any;
|
||||
const siliconflow = new SiliconFlowProvider({
|
||||
apiKey: config.ai.apiKey!,
|
||||
model: config.ai.model!,
|
||||
});
|
||||
const main = async () => {
|
||||
const usage = await siliconflow.getUsageInfo();
|
||||
console.log(usage);
|
||||
};
|
||||
|
||||
// main();
|
||||
const test = async () => {
|
||||
const result = await siliconflow.chat([
|
||||
{
|
||||
role: 'user',
|
||||
content: '你好',
|
||||
},
|
||||
]);
|
||||
console.log('result', result, 'result.choices[0].message.content', result.choices[0].message.content);
|
||||
return result.choices[0].message.content;
|
||||
};
|
||||
// test();
|
||||
22
assistant/src/test/provider/model-scope.ts
Normal file
22
assistant/src/test/provider/model-scope.ts
Normal file
@@ -0,0 +1,22 @@
|
||||
import { ProviderManager, SiliconFlowProvider, ModelScopeProvider } from '@kevisual/ai-center';
|
||||
|
||||
import { config } from 'dotenv';
|
||||
config();
|
||||
const providerTest = async () => {
|
||||
const providerConfig = { provider: 'ModelScope', model: 'Qwen/Qwen2.5-Coder-32B-Instruct', apiKey: process.env.MODEL_SCOPE_API_KEY };
|
||||
const provider = await ProviderManager.createProvider(providerConfig);
|
||||
const result = await provider.chat([{ role: 'user', content: '你好' }]);
|
||||
console.log(result);
|
||||
};
|
||||
|
||||
providerTest();
|
||||
|
||||
const modelScopeTest = async () => {
|
||||
const provider = new ModelScopeProvider({
|
||||
apiKey: process.env.MODEL_SCOPE_API_KEY,
|
||||
model: 'Qwen/Qwen2.5-Coder-32B-Instruct',
|
||||
});
|
||||
const result = await provider.chat([{ role: 'user', content: '你好' }]);
|
||||
console.log(result);
|
||||
};
|
||||
// modelScopeTest();
|
||||
Reference in New Issue
Block a user