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

chore: format

Browse Source
This commit is contained in:
tanzhenxin
2025-08-08 15:02:17 +08:00
parent 854c452580
commit 6f399c078a
4 changed files with 79 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
packages/core/src/core/__tests__/openaiTimeoutHandling.test.ts
View File

@@ -146,7 +146,9 @@ describe('OpenAIContentGenerator Timeout Handling', () => {
model: 'gpt-4',
};
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow(
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow(
/Request timeout after \d+s\. Try reducing input length or increasing timeout in config\./,
);
});
@@ -161,9 +163,9 @@ describe('OpenAIContentGenerator Timeout Handling', () => {
model: 'gpt-4',
};
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow(
'Invalid API key',
);
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow('Invalid API key');
});
it('should include troubleshooting tips for timeout errors', async () => {
@@ -199,7 +201,9 @@ describe('OpenAIContentGenerator Timeout Handling', () => {
model: 'gpt-4',
};
await expect(generator.generateContentStream(request, 'test-prompt-id')).rejects.toThrow(
await expect(
generator.generateContentStream(request, 'test-prompt-id'),
).rejects.toThrow(
/Streaming setup timeout after \d+s\. Try reducing input length or increasing timeout in config\./,
);
});
@@ -324,9 +328,9 @@ describe('OpenAIContentGenerator Timeout Handling', () => {
};
// Should not throw due to token counting failure
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow(
/Request timeout after \d+s/,
);
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow(/Request timeout after \d+s/);
});
});
});

63
packages/core/src/core/openaiContentGenerator.test.ts
View File

@@ -439,7 +439,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const stream = await generator.generateContentStream(request, 'test-prompt-id');
const stream = await generator.generateContentStream(
request,
'test-prompt-id',
);
const responses = [];
for await (const response of stream) {
responses.push(response);
@@ -528,7 +531,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const stream = await generator.generateContentStream(request, 'test-prompt-id');
const stream = await generator.generateContentStream(
request,
'test-prompt-id',
);
const responses = [];
for await (const response of stream) {
responses.push(response);
@@ -668,9 +674,9 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow(
'Invalid API key',
);
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow('Invalid API key');
});
it('should estimate tokens on error for telemetry', async () => {
@@ -872,7 +878,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const result = await generator.generateContent(request, 'test-prompt-id');
const result = await generator.generateContent(
request,
'test-prompt-id',
);
if (
result.candidates &&
result.candidates.length > 0 &&
@@ -969,7 +978,9 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow(
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow(
/Troubleshooting tips.*Reduce input length.*Increase timeout.*Check network/s,
);
});
@@ -985,9 +996,9 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow(
'Streaming setup failed',
);
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow('Streaming setup failed');
});
it('should handle timeout errors during streaming setup', async () => {
@@ -999,7 +1010,9 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
await expect(generator.generateContentStream(request, 'test-prompt-id')).rejects.toThrow(
await expect(
generator.generateContentStream(request, 'test-prompt-id'),
).rejects.toThrow(
/Streaming setup timeout troubleshooting.*Reduce input length/s,
);
});
@@ -1042,7 +1055,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const stream = await loggingGenerator.generateContentStream(request, 'test-prompt-id');
const stream = await loggingGenerator.generateContentStream(
request,
'test-prompt-id',
);
// Consume the stream and expect error
await expect(async () => {
@@ -1495,7 +1511,9 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
await expect(testGenerator.generateContent(request, 'test-prompt-id')).rejects.toThrow();
await expect(
testGenerator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow();
// Error logging should be suppressed
expect(consoleSpy).not.toHaveBeenCalledWith(
@@ -1519,7 +1537,9 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
await expect(generator.generateContent(request, 'test-prompt-id')).rejects.toThrow();
await expect(
generator.generateContent(request, 'test-prompt-id'),
).rejects.toThrow();
// Error logging should occur by default
expect(consoleSpy).toHaveBeenCalledWith(
@@ -1643,7 +1663,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const stream = await generator.generateContentStream(request, 'test-prompt-id');
const stream = await generator.generateContentStream(
request,
'test-prompt-id',
);
const responses = [];
for await (const response of stream) {
responses.push(response);
@@ -2039,7 +2062,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const stream = await loggingGenerator.generateContentStream(request, 'test-prompt-id');
const stream = await loggingGenerator.generateContentStream(
request,
'test-prompt-id',
);
const responses = [];
for await (const response of stream) {
responses.push(response);
@@ -2083,7 +2109,10 @@ describe('OpenAIContentGenerator', () => {
model: 'gpt-4',
};
const stream = await generator.generateContentStream(request, 'test-prompt-id');
const stream = await generator.generateContentStream(
request,
'test-prompt-id',
);
const responses = [];
for await (const response of stream) {
responses.push(response);

25
packages/core/src/qwen/qwenContentGenerator.test.ts
View File

@@ -154,7 +154,10 @@ describe('QwenContentGenerator', () => {
contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
};
const result = await qwenContentGenerator.generateContent(request, 'test-prompt-id');
const result = await qwenContentGenerator.generateContent(
request,
'test-prompt-id',
);
expect(result.text).toBe('Generated content');
expect(mockQwenClient.getAccessToken).toHaveBeenCalled();
@@ -171,7 +174,10 @@ describe('QwenContentGenerator', () => {
contents: [{ role: 'user', parts: [{ text: 'Hello stream' }] }],
};
const stream = await qwenContentGenerator.generateContentStream(request, 'test-prompt-id');
const stream = await qwenContentGenerator.generateContentStream(
request,
'test-prompt-id',
);
const chunks: string[] = [];
for await (const chunk of stream) {
@@ -238,7 +244,10 @@ describe('QwenContentGenerator', () => {
contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
};
const result = await qwenContentGenerator.generateContent(request, 'test-prompt-id');
const result = await qwenContentGenerator.generateContent(
request,
'test-prompt-id',
);
expect(result.text).toBe('Generated content');
expect(mockQwenClient.refreshAccessToken).toHaveBeenCalled();
@@ -545,7 +554,10 @@ describe('QwenContentGenerator', () => {
contents: [{ role: 'user', parts: [{ text: 'Hello' }] }],
};
const result = await qwenContentGenerator.generateContent(request, 'test-prompt-id');
const result = await qwenContentGenerator.generateContent(
request,
'test-prompt-id',
);
expect(result.text).toBe('Success after retry');
expect(mockGenerateContent).toHaveBeenCalledTimes(2);
@@ -795,7 +807,10 @@ describe('QwenContentGenerator', () => {
contents: [{ role: 'user', parts: [{ text: 'Test message' }] }],
};
const result = await qwenContentGenerator.generateContent(request, 'test-prompt-id');
const result = await qwenContentGenerator.generateContent(
request,
'test-prompt-id',
);
expect(result.text).toBe('Success after refresh');
expect(mockQwenClient.getAccessToken).toHaveBeenCalled();