fix(openai): rolled back response parsing for chat model

MaartenGr · Mar 15, 2024 · 5132c35 · 5132c35
1 parent ad85c14
commit 5132c35
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keybert/llm/_openai.py b/keybert/llm/_openai.py
@@ -178,7 +178,7 @@ def extract_keywords(self, documents: List[str], candidate_keywords: List[List[s
                     response = chat_completions_with_backoff(self.client, **kwargs)
                 else:
                     response = self.client.chat.completions.create(**kwargs)
-                keywords = response.choices[0].text.strip()
+                keywords = response.choices[0].message.content.strip()
 
             # Use a non-chat model
             else: