Austin/firecrawl
1
0
Fork 0
mirror of https://github.com/mendableai/firecrawl.git synced 2024-11-16 11:42:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Revert "Nick: improved map ranking algorithm"

Browse Source 
This reverts commit 7acd8d2edb.
This commit is contained in:
Nicolas 2024-10-21 16:11:32 -03:00
parent edac6850c6
commit d2344aa14b
3 changed files with 8 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
apps/api/src/controllers/v1/map.ts
View File

@ -2,7 +2,6 @@ import { Response } from "express";
import { v4 as uuidv4 } from "uuid";
import {
legacyCrawlerOptions,
LinkInfo,
mapRequestSchema,
RequestWithAuth,
} from "./types";
@ -110,10 +109,6 @@ export async function mapController(
mapResults = mapResults.slice(0, minumumCutoff);
}
let linkInfos: LinkInfo[] = [];
if (mapResults.length > 0) {
if (req.body.search) {
// Ensure all map results are first, maintaining their order
@ -122,12 +117,6 @@ export async function mapController(
...mapResults.slice(1).map((x) => x.url),
...links,
];
linkInfos = [
mapResults[0],
...mapResults.slice(1),
...links.map((x) => ({ url: x })),
]
} else {
mapResults.map((x) => {
links.push(x.url);
@ -139,7 +128,7 @@ export async function mapController(
if (req.body.search) {
const searchQuery = req.body.search.toLowerCase();
links = performCosineSimilarity(linkInfos, searchQuery);
links = performCosineSimilarity(links, searchQuery);
}
links = links

8
apps/api/src/controllers/v1/types.ts
View File

@ -478,11 +478,3 @@ export function legacyDocumentConverter(doc: any): Document {
},
};
}
export interface LinkInfo {
url: string;
title?: string;
description?: string;
}

15
apps/api/src/lib/map-cosine.ts
View File

@ -1,7 +1,6 @@
import { Logger } from "./logger";
import { LinkInfo } from "../controllers/v1/types";
export function performCosineSimilarity(links: LinkInfo[], searchQuery: string) {
export function performCosineSimilarity(links: string[], searchQuery: string) {
try {
// Function to calculate cosine similarity
const cosineSimilarity = (vec1: number[], vec2: number[]): number => {
@ -28,20 +27,20 @@ export function performCosineSimilarity(links: LinkInfo[], searchQuery: string)
// Calculate similarity scores
const similarityScores = links.map((link) => {
const linkText = `${link.url} ${link.title || ''} ${link.description || ''}`.trim();
const linkVector = textToVector(linkText);
const linkVector = textToVector(link);
const searchVector = textToVector(searchQuery);
return cosineSimilarity(linkVector, searchVector);
});
// Sort links based on similarity scores
const sortedLinks = links
// Sort links based on similarity scores and print scores
const a = links
.map((link, index) => ({ link, score: similarityScores[index] }))
.sort((a, b) => b.score - a.score);
return sortedLinks.map((item) => item.link.url);
links = a.map((item) => item.link);
return links;
} catch (error) {
Logger.error(`Error performing cosine similarity: ${error}`);
return links.map(link => link.url);
return links;
}
}