Skip to content

Commit d655c3c

Browse files
author
Julien Almarcha
committed
document manager note
1 parent 5f06633 commit d655c3c

File tree

1 file changed

+17
-9
lines changed

1 file changed

+17
-9
lines changed

app/helpers/_documentmanager.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -401,14 +401,22 @@ async def _query(
401401
response = await self._create_embeddings(input=[prompt], model_client=model_client)
402402
query_vector = response[0]
403403

404-
searches = await self.qdrant.search(
405-
method=method,
406-
collection_ids=collection_ids,
407-
query_prompt=prompt,
408-
query_vector=query_vector,
409-
k=k,
410-
rff_k=rff_k,
411-
score_threshold=score_threshold,
412-
)
404+
if method == SearchMethod.MULTIAGENT:
405+
# Appeler la classe MultiAgent
406+
searches = self.multiagent.search(
407+
qdrant=self.qdrant, collection_ids=collection_ids, query_vector=query_vector, k=k, rff_k=rff_k, score_threshold=score_threshold
408+
)
409+
410+
pass
411+
else:
412+
searches = await self.qdrant.search(
413+
method=method,
414+
collection_ids=collection_ids,
415+
query_prompt=prompt,
416+
query_vector=query_vector,
417+
k=k,
418+
rff_k=rff_k,
419+
score_threshold=score_threshold,
420+
)
413421

414422
return searches

0 commit comments

Comments
 (0)