chore(agents-api): fixed the limit params in chat for hybrid and text

julep-ai · Feb 26, 2025 · 59950fc · 59950fc
1 parent e8fa924
commit 59950fc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/agents-api/agents_api/queries/chat/gather_messages.py b/agents-api/agents_api/queries/chat/gather_messages.py
@@ -131,7 +131,7 @@ async def gather_messages(
                     developer_id=developer.id,
                     owners=owners,
                     embedding=query_embedding,
-                    k=recall_options.limit,
+                    k=recall_options.limit * 3 if recall_options.mmr_strength > 0 else recall_options.limit,
                     confidence=recall_options.confidence,
                     metadata_filter=recall_options.metadata_filter,
                     connection_pool=connection_pool,
@@ -142,7 +142,7 @@ async def gather_messages(
                     owners=owners,
                     text_query=query_text,
                     embedding=query_embedding,
-                    k=recall_options.limit,
+                    k=recall_options.limit * 3 if recall_options.mmr_strength > 0 else recall_options.limit,
                     search_language=search_language,
                     confidence=recall_options.confidence,
                     alpha=recall_options.alpha,