update functions
This commit is contained in:
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user