Update OpenAI API response (#213)

MaartenGr · Mar 21, 2024 · c886ee6 · c886ee6
1 parent f9e0e4a
commit c886ee6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keybert/llm/_openai.py b/keybert/llm/_openai.py
@@ -191,7 +191,7 @@ def extract_keywords(self, documents: List[str], candidate_keywords: List[List[s
                     response = completions_with_backoff(self.client, model=self.model, prompt=prompt, **self.generator_kwargs)
                 else:
                     response = self.client.completions.create(model=self.model, prompt=prompt, **self.generator_kwargs)
-                keywords = response.choices[0].message.content.strip()
+                keywords = response.choices[0].text.strip()
             keywords = [keyword.strip() for keyword in keywords.split(",")]
             all_keywords.append(keywords)