ishangsf/DeepCodeGeniusWeb-vscode
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #453 from devchat-ai/enhancement/combine-insert-retrieve-steps-#256

Browse Source 
Streamline Log Insertion by Combining Insert and Retrieve Actions
This commit is contained in:
boob.yang 2024-02-09 09:34:09 +08:00 committed by GitHub
commit dc58a93bed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
src/toolwrapper/devchat.ts
View File

@ -283,8 +283,7 @@ class DevChat {
const responseData = this.parseOutData(stdout, false);
let promptHash = "";
if (saveToLog) {
await this.logInsert(options.context, content, responseData.response, options.parent);
const logs = await this.log({"maxCount": 1});
const logs = await this.logInsert(options.context, content, responseData.response, options.parent);
assertValue(!logs || !logs.length, "Failed to insert devchat log");
promptHash = logs[0]["hash"];
}
@ -313,7 +312,7 @@ class DevChat {
}
}
async logInsert(contexts: string[] | undefined, request: string, response: string, parent: string | undefined): Promise<boolean> {
async logInsert(contexts: string[] | undefined, request: string, response: string, parent: string | undefined): Promise<LogEntry[]> {
try {
// build log data
const llmModelData = await ApiKeyManager.llmModel();
@ -357,11 +356,16 @@ class DevChat {
logger.channel()?.warn(`${stderr}`);
}
return true;
const logs = JSON.parse(stdout.trim()).reverse();
for (const log of logs) {
log.response = log.responses[0];
delete log.responses;
}
return logs;
} catch (error: any) {
logger.channel()?.error(`Failed to insert log: ${error.message}`);
logger.channel()?.show();
return false;
return [];
}
}