Fix #1666 (#1678)

MaartenGr · Dec 22, 2023 · 4572a99 · 4572a99
1 parent 1285f54
commit 4572a99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bertopic/representation/_openai.py b/bertopic/representation/_openai.py
@@ -227,7 +227,7 @@ def extract_topics(self,
                     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)
-                label = response.choices[0].message.content.strip()
+                label = response.choices[0].text.strip()
 
             updated_topics[topic] = [(label, 1)]