Skip to content

Commit 4e3a6b3

Browse files
committed
Merge branch 'main' into mermaid-diagrams-2
2 parents b090f04 + b08d821 commit 4e3a6b3

File tree

30 files changed

+1054
-274
lines changed

30 files changed

+1054
-274
lines changed

core/context/providers/_context-providers.vitest.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@ async function getContextProviderExtras(
5050
llm: config.modelsByRole.chat[0],
5151
reranker: config.selectedModelByRole.rerank,
5252
selectedCode: [],
53+
isInAgentMode: true,
5354
};
5455
}
5556

core/context/retrieval/pipelines/BaseRetrievalPipeline.ts

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,11 @@ Determine which tools should be used to answer this query. You should feel free
225225
.join("");
226226
const parsed = JSON.parse(responseContent);
227227
toolCalls = parsed.tools || [];
228-
console.log("retrieveWithTools", toolCalls);
229228
} catch (e) {
230-
console.log("Failed to parse tool selection response", e);
229+
console.log(
230+
`Failed to parse tool selection response: ${toolSelectionResponse.content}\n\n`,
231+
e,
232+
);
231233
return [];
232234
}
233235

core/context/retrieval/pipelines/RerankerRetrievalPipeline.ts

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -122,10 +122,7 @@ export default class RerankerRetrievalPipeline extends BaseRetrievalPipeline {
122122
results = results.slice(-this.options.nFinal);
123123
return results;
124124
} catch (e) {
125-
void this.options.ide.showToast(
126-
"warning",
127-
`Failed to rerank retrieval results\n${e}`,
128-
);
125+
console.warn(`Failed to rerank retrieval results\n${e}`);
129126
return chunks.slice(-this.options.nFinal);
130127
}
131128
}

core/context/retrieval/retrieval.ts

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -79,14 +79,19 @@ export async function retrieveContextItemsFromEmbeddings(
7979

8080
if (results.length === 0) {
8181
if (extras.config.disableIndexing) {
82-
void extras.ide.showToast("warning", "No results found.");
83-
return [];
84-
} else {
8582
void extras.ide.showToast(
8683
"warning",
87-
"No results found. If you think this is an error, re-index your codebase.",
84+
"No results found (Indexing disabled).",
8885
);
89-
// TODO - add "re-index" option to warning message which clears and reindexes codebase
86+
return [];
87+
}
88+
if (extras.isInAgentMode) {
89+
return [
90+
{
91+
...INSTRUCTIONS_BASE_ITEM,
92+
content: "No results were found. Try using other tools.",
93+
},
94+
];
9095
}
9196
return [];
9297
}

core/core.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1071,6 +1071,7 @@ export class Core {
10711071
query: string;
10721072
fullInput: string;
10731073
selectedCode: RangeInFile[];
1074+
isInAgentMode: boolean;
10741075
}>,
10751076
) => {
10761077
const { config } = await this.configHandler.loadConfig();
@@ -1115,6 +1116,7 @@ export class Core {
11151116
reranker: config.selectedModelByRole.rerank,
11161117
fetch: (url, init) =>
11171118
fetchwithRequestOptions(url, init, config.requestOptions),
1119+
isInAgentMode: msg.data.isInAgentMode,
11181120
});
11191121

11201122
void Telemetry.capture(

0 commit comments

Comments
 (0)