update functions

This commit is contained in:
Ayomide
2025-07-14 23:41:31 +01:00
parent 0b5a7218b0
commit 97a3b710c3
7 changed files with 580 additions and 75 deletions
+6 -2
View File
@@ -10,15 +10,19 @@ class EmbeddingGenerator:
response = self.client.embed(
texts=[text],
model=Config.EMBED_MODEL,
input_type="document"
input_type="search_document"
)
return response.embeddings[0]
def rerank_issues(self, issues: list, query: str, top_n: int = 5):
# Handle empty issues list
if not issues:
return []
response = self.client.rerank(
query=query,
documents=issues,
top_n=top_n,
top_n=min(top_n, len(issues)),
model=Config.RERANK_MODEL
)
return [result.document for result in response.results]