Fix bug relating to keyword extraction chain

catalystneuro · Oct 13, 2023 · decf8c2 · decf8c2 · jaiamin · Oct 13, 2023
1 parent d52e459
commit decf8c2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/rest/clients/openai.py b/rest/clients/openai.py
@@ -92,6 +92,8 @@ def keywords_extraction(self, user_input: str, model: str = "gpt-3.5-turbo"):
         )
         chain = create_extraction_chain(schema, llm)
         keywords_extracted = list(chain.run(user_input))
+        if any(isinstance(item, str) for item in keywords_extracted):
+            keywords_extracted = [{key: ""} for key in schema["properties"]]
         return self.prepare_keywords_for_semantic_search(keywords_extracted)