run-query.ts

type QueryResult = {
	score: number,
	func: FunctionInfo
}
export async function runQuery(
	query: string,
	files: FileInfo[]
): Promise<QueryResult[]> {
	// convert input query to embedding
	const queryEmbedding = await createEmbedding(query);
	const similarityScores: QueryResult[] = [];
	for (const file of files) {
		for (const func of file.functions) {
			// calclulate similarity score
			const similarity = cosineSimilarity(queryEmbedding, func.embedding);
			similarityScores.push({
				score: similarity, 
				func
			})
		}
	}
	// sort results by score
	const sorted = similarityScores.sort((a, b) => b.score - a.score);
 
	// return the top 10
	return similarityScores.slice(0, 10)
}