elias/core
1
0
Fork 0
mirror of https://github.com/eliasstepanik/core.git synced 2026-01-25 10:58:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: increase reranking score threshold

Browse Source
This commit is contained in:
Manoj 2025-09-19 10:31:01 +05:30
parent a0b3128329
commit 963bfcbb78
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
apps/webapp/app/env.server.ts
View File

@ -89,6 +89,7 @@ const EnvironmentSchema = z.object({
EMBEDDING_MODEL_SIZE: z.string().default("1024"), EMBEDDING_MODEL_SIZE: z.string().default("1024"),
OLLAMA_URL: z.string().optional(), OLLAMA_URL: z.string().optional(),
COHERE_API_KEY: z.string().optional(), COHERE_API_KEY: z.string().optional(),
COHERE_SCORE_THRESHOLD: z.string().default("0.3"),
}); });
export type Environment = z.infer<typeof EnvironmentSchema>; export type Environment = z.infer<typeof EnvironmentSchema>;

3
apps/webapp/app/services/search/rerank.ts
View File

@ -4,6 +4,7 @@ import { type CoreMessage } from "ai";
import { makeModelCall } from "~/lib/model.server"; import { makeModelCall } from "~/lib/model.server";
import { logger } from "../logger.service"; import { logger } from "../logger.service";
import { CohereClientV2 } from "cohere-ai"; import { CohereClientV2 } from "cohere-ai";
import { env } from "~/env.server";
// Utility function to safely convert BigInt values to Number // Utility function to safely convert BigInt values to Number
function safeNumber(value: any): number { function safeNumber(value: any): number {
@ -512,7 +513,7 @@ export async function applyCohereReranking(
cohereScore: result.relevanceScore, cohereScore: result.relevanceScore,
cohereRank: index + 1, cohereRank: index + 1,
})) }))
.filter((result) => result.cohereScore >= 0.1); .filter((result) => result.cohereScore >= Number(env.COHERE_SCORE_THRESHOLD));
const responseTime = Date.now() - startTime; const responseTime = Date.now() - startTime;
logger.info( logger.info(