fix(openai): changed openai api response parsing

MaartenGr · Mar 13, 2024 · ad85c14 · ad85c14
1 parent dcf31dd
commit ad85c14
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/keybert/llm/_openai.py b/keybert/llm/_openai.py
@@ -178,15 +178,15 @@ 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].message.content.strip()
+                keywords = response.choices[0].text.strip()
 
             # Use a non-chat model
             else:
                 if self.exponential_backoff:
                     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)