diff --git a/src/LLMProviders/BedrockChatModel.ts b/src/LLMProviders/BedrockChatModel.ts index 22c8fabf..f8dabc7a 100644 --- a/src/LLMProviders/BedrockChatModel.ts +++ b/src/LLMProviders/BedrockChatModel.ts @@ -391,13 +391,14 @@ export class BedrockChatModel extends BaseChatModel const fallback = await this._generate(messages, options, runManager); const fallbackText = fallback.generations[0]?.text ?? ""; if (fallbackText) { + const fallbackMetadata: Record = fallback.llmOutput ?? {}; yield new ChatGenerationChunk({ message: new AIMessageChunk({ content: fallbackText, - response_metadata: fallback.llmOutput ?? {}, + response_metadata: fallbackMetadata, }), text: fallbackText, - generationInfo: fallback.llmOutput ?? {}, + generationInfo: fallbackMetadata, }); } } @@ -1251,7 +1252,7 @@ export class BedrockChatModel extends BaseChatModel const systemPrompts: string[] = []; messages.forEach((message) => { - const messageType = message.getType(); + const messageType = message.type; // Handle system messages (always text-only) if (messageType === "system") { diff --git a/src/search/dbOperations.ts b/src/search/dbOperations.ts index 066b98b5..484dbc22 100644 --- a/src/search/dbOperations.ts +++ b/src/search/dbOperations.ts @@ -368,7 +368,7 @@ export class DBOperations { }; } - async upsert(docToSave: any): Promise { + async upsert(docToSave: OramaDocument): Promise { if (!this.oramaDb) throw new Error("DB not initialized"); const db = this.oramaDb; @@ -400,7 +400,7 @@ export class DBOperations { ); this.markUnsavedChanges(); - return docToSave as unknown; + return docToSave; } catch (insertErr) { logError( `Failed to ${existingDoc.hits.length > 0 ? "update" : "insert"} document ${docToSave.id}:`, diff --git a/src/search/v3/utils/ScoreNormalizer.ts b/src/search/v3/utils/ScoreNormalizer.ts index a4e348ff..7e736e9c 100644 --- a/src/search/v3/utils/ScoreNormalizer.ts +++ b/src/search/v3/utils/ScoreNormalizer.ts @@ -1,4 +1,4 @@ -import { NoteIdRank } from "../interfaces"; +import { NoteIdRank, SearchExplanation } from "../interfaces"; /** * Configuration for score normalization @@ -29,7 +29,11 @@ export class ScoreNormalizer { /** * Update explanation with normalized scores */ - private updateExplanation(explanation: any, originalScore: number, normalizedScore: number): any { + private updateExplanation( + explanation: SearchExplanation | undefined, + originalScore: number, + normalizedScore: number + ): SearchExplanation | undefined { if (!explanation) return undefined; return {