revert: topp & temperature default velue to none

This commit is contained in:
koalazf.99
2025-10-28 13:00:10 +08:00
parent 7ccba75621
commit 354c85bcff
2 changed files with 40 additions and 8 deletions

View File

@@ -65,10 +65,7 @@ export interface GenerateJsonOptions {
*/ */
export class BaseLlmClient { export class BaseLlmClient {
// Default configuration for utility tasks // Default configuration for utility tasks
private readonly defaultUtilityConfig: GenerateContentConfig = { private readonly defaultUtilityConfig: GenerateContentConfig = {};
temperature: 0,
topP: 1,
};
constructor( constructor(
private readonly contentGenerator: ContentGenerator, private readonly contentGenerator: ContentGenerator,

View File

@@ -149,10 +149,7 @@ const COMPRESSION_PRESERVE_THRESHOLD = 0.3;
export class GeminiClient { export class GeminiClient {
private chat?: GeminiChat; private chat?: GeminiChat;
private readonly generateContentConfig: GenerateContentConfig = { private readonly generateContentConfig: GenerateContentConfig;
temperature: 0,
topP: 1,
};
private sessionTurnCount = 0; private sessionTurnCount = 0;
private readonly loopDetector: LoopDetectionService; private readonly loopDetector: LoopDetectionService;
@@ -169,6 +166,44 @@ export class GeminiClient {
constructor(private readonly config: Config) { constructor(private readonly config: Config) {
this.loopDetector = new LoopDetectionService(config); this.loopDetector = new LoopDetectionService(config);
this.lastPromptId = this.config.getSessionId(); this.lastPromptId = this.config.getSessionId();
this.generateContentConfig = this.buildDefaultGenerateContentConfig();
}
private buildDefaultGenerateContentConfig(): GenerateContentConfig {
const samplingParams =
this.config.getContentGeneratorConfig()?.samplingParams;
if (!samplingParams) {
return {};
}
const config: GenerateContentConfig = {};
if (samplingParams.temperature !== undefined) {
config.temperature = samplingParams.temperature;
}
if (samplingParams.top_p !== undefined) {
config.topP = samplingParams.top_p;
}
if (samplingParams.top_k !== undefined) {
config.topK = samplingParams.top_k;
}
if (samplingParams.max_tokens !== undefined) {
config.maxOutputTokens = samplingParams.max_tokens;
}
if (samplingParams.presence_penalty !== undefined) {
config.presencePenalty = samplingParams.presence_penalty;
}
if (samplingParams.frequency_penalty !== undefined) {
config.frequencyPenalty = samplingParams.frequency_penalty;
}
return config;
} }
async initialize() { async initialize() {