abbyswag/qwen-code
1
0
Fork 0
mirror of https://github.com/QwenLM/qwen-code.git synced 2025-12-19 09:33:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix CODE_ASSIST_ENDPOINT env var. (#2712)

Browse Source
This commit is contained in:
Tommaso Sciortino
2025-06-30 15:41:14 -07:00
committed by GitHub
parent 36e099ac22
commit 505a5d617b
2 changed files with 28 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
packages/core/src/code_assist/server.test.ts
View File

@@ -35,14 +35,14 @@ describe('CodeAssistServer', () => {
],
},
};
vi.spyOn(server, 'callEndpoint').mockResolvedValue(mockResponse);
vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
const response = await server.generateContent({
model: 'test-model',
contents: [{ role: 'user', parts: [{ text: 'request' }] }],
});
expect(server.callEndpoint).toHaveBeenCalledWith(
expect(server.requestPost).toHaveBeenCalledWith(
'generateContent',
expect.any(Object),
undefined,
@@ -72,7 +72,7 @@ describe('CodeAssistServer', () => {
},
};
})();
vi.spyOn(server, 'streamEndpoint').mockResolvedValue(mockResponse);
vi.spyOn(server, 'requestStreamingPost').mockResolvedValue(mockResponse);
const stream = await server.generateContentStream({
model: 'test-model',
@@ -80,7 +80,7 @@ describe('CodeAssistServer', () => {
});
for await (const res of stream) {
expect(server.streamEndpoint).toHaveBeenCalledWith(
expect(server.requestStreamingPost).toHaveBeenCalledWith(
'streamGenerateContent',
expect.any(Object),
undefined,
@@ -96,7 +96,7 @@ describe('CodeAssistServer', () => {
name: 'operations/123',
done: true,
};
vi.spyOn(server, 'callEndpoint').mockResolvedValue(mockResponse);
vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
const response = await server.onboardUser({
tierId: 'test-tier',
@@ -104,7 +104,7 @@ describe('CodeAssistServer', () => {
metadata: {},
});
expect(server.callEndpoint).toHaveBeenCalledWith(
expect(server.requestPost).toHaveBeenCalledWith(
'onboardUser',
expect.any(Object),
);
@@ -117,13 +117,13 @@ describe('CodeAssistServer', () => {
const mockResponse = {
// TODO: Add mock response
};
vi.spyOn(server, 'callEndpoint').mockResolvedValue(mockResponse);
vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
const response = await server.loadCodeAssist({
metadata: {},
});
expect(server.callEndpoint).toHaveBeenCalledWith(
expect(server.requestPost).toHaveBeenCalledWith(
'loadCodeAssist',
expect.any(Object),
);
@@ -136,7 +136,7 @@ describe('CodeAssistServer', () => {
const mockResponse = {
totalTokens: 100,
};
vi.spyOn(server, 'callEndpoint').mockResolvedValue(mockResponse);
vi.spyOn(server, 'requestPost').mockResolvedValue(mockResponse);
const response = await server.countTokens({
model: 'test-model',