fix(api): add missing user_agent argument

astronomer · Nov 22, 2023 · 8d52268 · 8d52268
1 parent 684ba92
commit 8d52268
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/api/ask_astro/chains/answer_question.py b/api/ask_astro/chains/answer_question.py
@@ -68,8 +68,7 @@ def _get_docs(
     ):
         docs = super()._get_docs(question=question, inputs=inputs, run_manager=run_manager)
         if CohereConfig.cohere_api_key:
-            # compressor = CohereRerank(top_n=CohereConfig.top_n)
-            compressor = CohereRerank()
+            compressor = CohereRerank(top_n=CohereConfig.top_n, user_agent="langchain")
             return compressor.compress_documents(docs, question)
         return docs