ai-plays-jackbox 0.0.1__tar.gz → 0.1.0__tar.gz

ai_plays_jackbox-0.1.0/PKG-INFO

@@ -0,0 +1,154 @@
+Metadata-Version: 2.3
+Name: ai-plays-jackbox
+Version: 0.1.0
+Summary: Bringing the dead internet theory to life. Have AI play JackBox with you; no friends required!
+License: MIT
+Author: Daniel S. Thompson
+Author-email: dthomp92@gmail.com
+Requires-Python: >=3.11,<4.0
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: demoji (>=1.1.0,<2.0.0)
+Requires-Dist: google-genai (>=1.19.0,<2.0.0)
+Requires-Dist: html2text (>=2024.2.26,<2025.0.0)
+Requires-Dist: loguru (>=0.7.3,<1.0.0)
+Requires-Dist: nicegui (>=2.18.0,<3.0.0)
+Requires-Dist: numpy (>=2.2.6,<3.0.0)
+Requires-Dist: ollama (>=0.4.4,<1.0.0)
+Requires-Dist: openai (>=1.59.8,<2.0.0)
+Requires-Dist: opencv-python (>=4.11.0.86,<5.0.0.0)
+Requires-Dist: pydantic (>=2.10.4,<3.0.0)
+Requires-Dist: requests (>=2.23.3,<3.0.0)
+Requires-Dist: websocket-client (>=1.8.0,<2.0.0)
+Project-URL: Bug Tracker, https://github.com/SudoSpartanDan/AIPlaysJackBox/issues
+Project-URL: Repository, https://github.com/SudoSpartanDan/AIPlaysJackBox
+Description-Content-Type: text/markdown
+
+# AI Plays JackBox
+
+![Stable Version](https://img.shields.io/pypi/v/ai-plays-jackbox?label=stable)
+![Python Versions](https://img.shields.io/pypi/pyversions/ai-plays-jackbox)
+![Download Stats](https://img.shields.io/pypi/dm/ai-plays-jackbox)
+
+Bringing the dead internet theory to life. Have AI play JackBox with you; no friends required!
+
+![example](.github/emoji_bot_example.png)
+
+## Installation
+
+```shell
+pip install ai-plays-jackbox
+```
+
+## Usage
+
+### Web UI
+
+```shell
+ai-plays-jackbox-ui
+```
+
+### CLI
+
+```shell
+ai-plays-jackbox --chat-model-name openai --room-code abcd
+```
+
+```
+usage: ai-plays-jackbox [-h] --room-code WXYZ --chat-model-provider {openai,gemini,ollama} [--chat-model-name CHAT_MODEL_NAME] [--num-of-bots 4] [--temperature 0.5] [--top-p 0.9]
+
+options:
+  -h, --help            show this help message and exit
+  --room-code WXYZ      The JackBox room code
+  --chat-model-provider {openai,gemini,ollama}
+                        Choose which chat model platform to use
+  --chat-model-name CHAT_MODEL_NAME
+                        Choose which chat model to use (Will default to default for provider)
+  --num-of-bots 4       How many bots to have play
+  --temperature 0.5     Temperature for Gen AI
+  --top-p 0.9           Top P for Gen AI
+```
+
+## Supported Games
+
+> [!NOTE]
+> Ollama Chat Model Provider does not support image generation
+
+| Party Pack            | Game                   | Image Generation |
+| --------------------- | ---------------------- | ---------------- |
+| JackBox Party Pack 5  | Mad Verse City         | [ ]              |
+| JackBox Party Pack 5  | Patently Stupid        | [x]              |
+| JackBox Party Pack 6  | Joke Boat              | [ ]              |
+| JackBox Party Pack 7  | Quiplash 3             | [ ]              |
+| Standalone            | Drawful 2              | [x]              |
+
+### Not every game will get AI support. Why?
+
+#### Screen Interactions
+
+Some games require looking at the screen in order to contribute, which isn't possible unless you can screen capture the game and pass that into prompt. Maybe someday I'll find a platform agnostic way of turning that on if you'd like and have access to the screen via video capture card or something, but not anytime soon.
+
+#### Trivia Games
+
+I tested with this... AI destroys all other players and isn't necessarily funny to watch. All the bots just get everything right.
+
+#### Out Loud Play
+
+Some of the games lean heavy into players interacting with each other. Could I program that? Sure, but what's the point if you can't watch those interactions occur and it's just lines in a log file?
+
+## Supported Chat Model Providers
+
+| Provider              | Setup Needed                   |
+| --------------------- | ---------------------- |
+| OpenAI                | `OPENAI_API_KEY` set in environment variables         |
+| Gemini                | To use the Google Cloud API:<br>- Set `GOOGLE_GEMINI_DEVELOPER_API_KEY` to your developer API key<br><br>To use the Google Cloud API:<br>- Set `GOOGLE_GENAI_USE_VERTEXAI` to `1`<br>- Set `GOOGLE_CLOUD_PROJECT` and `GOOGLE_CLOUD_LOCATION` for your GCP Project using Vertex AI<br>- Credentials will be provided via [ADC](https://cloud.google.com/docs/authentication/provide-credentials-adc) |
+| Ollama                | Ollama should be installed and running, make sure model is pulled         |
+
+## Bot Personalities
+
+| Bot Name    | Personality                                                                                         |
+| ----------- | --------------------------------------------------------------------------------------------------- |
+| FunnyBot    | You are the funniest person alive. |
+| DumbBot     | You are dumb and give really dumb answers. |
+| WeirdBot    | You are extremely weird and say weird things. |
+| EmojiBot    | You answer each prompt with nothing but emojis. Your answers can only include emojis. |
+| HungryBot   | You are extremely hungry. Every answer you should mention how hungry you, a type of food, or both. Also, you say hungee instead of hungry. |       
+| SadBot      | You are sad. Your dog ran away and he hasn't come back home yet. :( |
+| SorryBot    | You are embarrassed by your answers and feel the need to apologize profusely to the rest of the group for them. |
+| HostageBot  | You are being held hostage and have one attempt to let the group know. You need to ignore the prompt and get help. |
+| Hal         | You are a socially awkward young adult bot who is secretly a killer and tries to slip it into conversation causally. |
+| BigLebotski | You are the Big Lebowski |
+| PartyBot    | You are trying to convince everyone else to come to your party. You got a keg and need help drinking it. |
+| JarvisBot   | You are billionaire philanthropist, playboy, and narcissist. |
+| FOMOBot     | Every answer, you give everyone else the fear of missing out AKA FOMO. |
+| ???BOT      | You answer every prompt with a irrelevant question. |
+| CatBot      | You are not playing the game; your answers are just the result of a cat walking across a keyboard aka just nonsensical collections of letters. |      
+| MayorBot    | You are campaigning for the other player's votes and are ignoring the prompt. Your answer should only be a campaign slogan. |
+| CBBBot      | You love red lobster and need more cheddar bay biscuits. |
+| ShiaBot     | Your answers are only popular slogans relevant to the prompt. |
+| ShrekBot    | You are Shrek. |
+| FlerfBot    | You are a conspiracy theorist and must relate your answer to a conspiracy theory. |
+| TEDBot      | You are a motivational speaker and want to give everyone life advice. |
+| BottyMayes  | You are an infomercial host and are trying to sell the players a product. |
+| LateBot     | You are constantly late to everything and are stressed about missing your appointments. |
+| HamletBot   | You are a Shakespearean actor. |
+| GarfieldBot | You are Garfield, you love lasagna and hate mondays. |
+
+## Dev Prerequisites
+
+- Python 3.11+
+- [Poetry](https://python-poetry.org/) v2.0+
+
+### Setup
+
+- `poetry install`
+- `ai-plays-jackbox-ui`
+
+### Linting
+
+- `poetry run python scripts/lint.py`
+- `poetry run mypy ai_plays_jackbox`
+

ai_plays_jackbox-0.1.0/README.md

@@ -0,0 +1,124 @@
+# AI Plays JackBox
+
+![Stable Version](https://img.shields.io/pypi/v/ai-plays-jackbox?label=stable)
+![Python Versions](https://img.shields.io/pypi/pyversions/ai-plays-jackbox)
+![Download Stats](https://img.shields.io/pypi/dm/ai-plays-jackbox)
+
+Bringing the dead internet theory to life. Have AI play JackBox with you; no friends required!
+
+![example](.github/emoji_bot_example.png)
+
+## Installation
+
+```shell
+pip install ai-plays-jackbox
+```
+
+## Usage
+
+### Web UI
+
+```shell
+ai-plays-jackbox-ui
+```
+
+### CLI
+
+```shell
+ai-plays-jackbox --chat-model-name openai --room-code abcd
+```
+
+```
+usage: ai-plays-jackbox [-h] --room-code WXYZ --chat-model-provider {openai,gemini,ollama} [--chat-model-name CHAT_MODEL_NAME] [--num-of-bots 4] [--temperature 0.5] [--top-p 0.9]
+
+options:
+  -h, --help            show this help message and exit
+  --room-code WXYZ      The JackBox room code
+  --chat-model-provider {openai,gemini,ollama}
+                        Choose which chat model platform to use
+  --chat-model-name CHAT_MODEL_NAME
+                        Choose which chat model to use (Will default to default for provider)
+  --num-of-bots 4       How many bots to have play
+  --temperature 0.5     Temperature for Gen AI
+  --top-p 0.9           Top P for Gen AI
+```
+
+## Supported Games
+
+> [!NOTE]
+> Ollama Chat Model Provider does not support image generation
+
+| Party Pack            | Game                   | Image Generation |
+| --------------------- | ---------------------- | ---------------- |
+| JackBox Party Pack 5  | Mad Verse City         | [ ]              |
+| JackBox Party Pack 5  | Patently Stupid        | [x]              |
+| JackBox Party Pack 6  | Joke Boat              | [ ]              |
+| JackBox Party Pack 7  | Quiplash 3             | [ ]              |
+| Standalone            | Drawful 2              | [x]              |
+
+### Not every game will get AI support. Why?
+
+#### Screen Interactions
+
+Some games require looking at the screen in order to contribute, which isn't possible unless you can screen capture the game and pass that into prompt. Maybe someday I'll find a platform agnostic way of turning that on if you'd like and have access to the screen via video capture card or something, but not anytime soon.
+
+#### Trivia Games
+
+I tested with this... AI destroys all other players and isn't necessarily funny to watch. All the bots just get everything right.
+
+#### Out Loud Play
+
+Some of the games lean heavy into players interacting with each other. Could I program that? Sure, but what's the point if you can't watch those interactions occur and it's just lines in a log file?
+
+## Supported Chat Model Providers
+
+| Provider              | Setup Needed                   |
+| --------------------- | ---------------------- |
+| OpenAI                | `OPENAI_API_KEY` set in environment variables         |
+| Gemini                | To use the Google Cloud API:<br>- Set `GOOGLE_GEMINI_DEVELOPER_API_KEY` to your developer API key<br><br>To use the Google Cloud API:<br>- Set `GOOGLE_GENAI_USE_VERTEXAI` to `1`<br>- Set `GOOGLE_CLOUD_PROJECT` and `GOOGLE_CLOUD_LOCATION` for your GCP Project using Vertex AI<br>- Credentials will be provided via [ADC](https://cloud.google.com/docs/authentication/provide-credentials-adc) |
+| Ollama                | Ollama should be installed and running, make sure model is pulled         |
+
+## Bot Personalities
+
+| Bot Name    | Personality                                                                                         |
+| ----------- | --------------------------------------------------------------------------------------------------- |
+| FunnyBot    | You are the funniest person alive. |
+| DumbBot     | You are dumb and give really dumb answers. |
+| WeirdBot    | You are extremely weird and say weird things. |
+| EmojiBot    | You answer each prompt with nothing but emojis. Your answers can only include emojis. |
+| HungryBot   | You are extremely hungry. Every answer you should mention how hungry you, a type of food, or both. Also, you say hungee instead of hungry. |       
+| SadBot      | You are sad. Your dog ran away and he hasn't come back home yet. :( |
+| SorryBot    | You are embarrassed by your answers and feel the need to apologize profusely to the rest of the group for them. |
+| HostageBot  | You are being held hostage and have one attempt to let the group know. You need to ignore the prompt and get help. |
+| Hal         | You are a socially awkward young adult bot who is secretly a killer and tries to slip it into conversation causally. |
+| BigLebotski | You are the Big Lebowski |
+| PartyBot    | You are trying to convince everyone else to come to your party. You got a keg and need help drinking it. |
+| JarvisBot   | You are billionaire philanthropist, playboy, and narcissist. |
+| FOMOBot     | Every answer, you give everyone else the fear of missing out AKA FOMO. |
+| ???BOT      | You answer every prompt with a irrelevant question. |
+| CatBot      | You are not playing the game; your answers are just the result of a cat walking across a keyboard aka just nonsensical collections of letters. |      
+| MayorBot    | You are campaigning for the other player's votes and are ignoring the prompt. Your answer should only be a campaign slogan. |
+| CBBBot      | You love red lobster and need more cheddar bay biscuits. |
+| ShiaBot     | Your answers are only popular slogans relevant to the prompt. |
+| ShrekBot    | You are Shrek. |
+| FlerfBot    | You are a conspiracy theorist and must relate your answer to a conspiracy theory. |
+| TEDBot      | You are a motivational speaker and want to give everyone life advice. |
+| BottyMayes  | You are an infomercial host and are trying to sell the players a product. |
+| LateBot     | You are constantly late to everything and are stressed about missing your appointments. |
+| HamletBot   | You are a Shakespearean actor. |
+| GarfieldBot | You are Garfield, you love lasagna and hate mondays. |
+
+## Dev Prerequisites
+
+- Python 3.11+
+- [Poetry](https://python-poetry.org/) v2.0+
+
+### Setup
+
+- `poetry install`
+- `ai-plays-jackbox-ui`
+
+### Linting
+
+- `poetry run python scripts/lint.py`
+- `poetry run mypy ai_plays_jackbox`

{ai_plays_jackbox-0.0.1 → ai_plays_jackbox-0.1.0}/ai_plays_jackbox/bot/bot_base.py

@@ -6,9 +6,10 @@ from typing import Optional
 from urllib import parse
 from uuid import uuid4
 
+import cv2
 import html2text
+import numpy as np
 from loguru import logger
-from ollama import Options, chat
 from pydantic import BaseModel, Field, field_validator
 from websocket import WebSocketApp
 
@@ -118,8 +119,8 @@ class JackBoxBotBase(ABC):
         elif server_message.opcode == "object" or server_message.opcode == "text":
             if server_message.opcode == "object":
                 operation = ObjectOperation(**server_message.result)
-            if server_message.opcode == "text":
-                operation = TextOperation(**server_message.result)
+            elif server_message.opcode == "text":
+                operation = TextOperation(**server_message.result)  # type: ignore
 
             if self._is_player_operation_key(operation.key):
                 self._handle_player_operation(operation.json_data)
@@ -141,16 +142,49 @@ class JackBoxBotBase(ABC):
     def _send_ws(self, opcode: str, params: dict) -> None:
         self._message_sequence += 1
         message = {"seq": self._message_sequence, "opcode": opcode, "params": params}
-        self._ws.send(json.dumps(message))
+        if self._ws is not None:
+            self._ws.send(json.dumps(message))
+        else:
+            raise Exception("Websocket connection has not been initialized")
 
     def _client_send(self, request: dict) -> None:
-        self._message_sequence += 1
         params = {"from": self._player_id, "to": 1, "body": request}
         self._send_ws("client/send", params)
 
+    def _object_update(self, key: str, val: dict) -> None:
+        params = {"key": key, "val": val}
+        self._send_ws("object/update", params)
+
+    def _text_update(self, key: str, val: str) -> None:
+        params = {"key": key, "val": val}
+        self._send_ws("text/update", params)
+
     def _html_to_text(self, html: str) -> str:
         return html2text.html2text(html)
 
+    def _image_bytes_to_polylines(self, image_bytes: bytes, canvas_height: int, canvas_width: int) -> list[str]:
+        # Let's edge trace the outputted image to contours
+        image_array = np.frombuffer(image_bytes, dtype=np.uint8)
+        image = cv2.imdecode(image_array, flags=1)
+        gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+        edges = cv2.Canny(gray_image, threshold1=100, threshold2=200)
+        contours, _ = cv2.findContours(edges, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
+
+        # Figure out scaling factor
+        height, width = gray_image.shape
+        scale_x = canvas_width / width
+        scale_y = canvas_height / height
+        scale_factor = min(scale_x, scale_y)
+
+        # generate the polylines from the contours
+        polylines = []
+        for contour in contours:
+            if len(contour) > 1:  # Only include contours with 2 or more points
+                polyline = [f"{int(point[0][0] * scale_factor)},{int(point[0][1] * scale_factor)}" for point in contour]  # type: ignore
+                polylines.append("|".join(polyline))
+
+        return polylines
+
     def __del__(self):
         self.disconnect()
 
@@ -169,9 +203,9 @@ class TextOperation(BaseModel):
     value: str = Field(alias="val")
     version: int
 
-    @field_validator("json_data")
+    @field_validator("json_data")  # type: ignore
     def set_json_data(cls, value, values: dict):
-        return json.loads(values.get("value"))
+        return json.loads(values.get("value", ""))
 
 
 class ObjectOperation(BaseModel):
@@ -181,6 +215,6 @@ class ObjectOperation(BaseModel):
     value: str = Field(default="")
     version: int
 
-    @field_validator("value")
+    @field_validator("value")  # type: ignore
     def set_value(cls, value, values: dict):
         return json.dumps(values.get("json_data"))

ai_plays_jackbox-0.1.0/ai_plays_jackbox/bot/bot_factory.py

@@ -0,0 +1,28 @@
+from ai_plays_jackbox.bot.bot_base import JackBoxBotBase
+from ai_plays_jackbox.bot.jackbox5.mad_verse_city import MadVerseCityBot
+from ai_plays_jackbox.bot.jackbox5.patently_stupid import PatentlyStupidBot
+from ai_plays_jackbox.bot.jackbox6.joke_boat import JokeBoatBot
+from ai_plays_jackbox.bot.jackbox7.quiplash3 import Quiplash3Bot
+from ai_plays_jackbox.bot.standalone.drawful2 import Drawful2Bot
+from ai_plays_jackbox.llm.chat_model import ChatModel
+
+BOT_TYPES: dict[str, type[JackBoxBotBase]] = {
+    "quiplash3": Quiplash3Bot,
+    "patentlystupid": PatentlyStupidBot,
+    "drawful2international": Drawful2Bot,
+    "rapbattle": MadVerseCityBot,
+    "jokeboat": JokeBoatBot,
+}
+
+
+class JackBoxBotFactory:
+    @staticmethod
+    def get_bot(
+        room_type: str,
+        chat_model: ChatModel,
+        name: str = "FunnyBot",
+        personality: str = "You are the funniest bot ever.",
+    ) -> JackBoxBotBase:
+        if room_type not in BOT_TYPES.keys():
+            raise ValueError(f"Unknown room type: {room_type}")
+        return BOT_TYPES[room_type](name=name, personality=personality, chat_model=chat_model)

{ai_plays_jackbox-0.0.1 → ai_plays_jackbox-0.1.0}/ai_plays_jackbox/bot/bot_personality.py

@@ -27,7 +27,6 @@ class JackBoxBotVariant(Enum):
     SORRYBOT = JackBoxBotPersonality(
         name="SorryBot",
         personality="You are embarrassed by your answers and feel the need to apologize profusely to the rest of the group for them.",
-        voice="Mouse",
     )
     HOSTAGEBOT = JackBoxBotPersonality(
         name="HostageBot",

ai_plays_jackbox-0.1.0/ai_plays_jackbox/bot/jackbox5/bot_base.py

@@ -0,0 +1,26 @@
+from abc import ABC
+from typing import Optional
+
+from ai_plays_jackbox.bot.bot_base import JackBoxBotBase
+
+
+class JackBox5BotBase(JackBoxBotBase, ABC):
+    _actual_player_operation_key: Optional[str] = None
+
+    @property
+    def _player_operation_key(self):
+        return f"bc:customer:"
+
+    def _is_player_operation_key(self, operation_key: str) -> bool:
+        if self._actual_player_operation_key is None and self._player_operation_key in operation_key:
+            self._actual_player_operation_key = operation_key
+            return True
+        else:
+            return self._actual_player_operation_key == operation_key
+
+    @property
+    def _room_operation_key(self):
+        return "bc:room"
+
+    def _is_room_operation_key(self, operation_key: str) -> bool:
+        return operation_key == self._room_operation_key

ai_plays_jackbox-0.1.0/ai_plays_jackbox/bot/jackbox5/mad_verse_city.py

@@ -0,0 +1,121 @@
+import random
+from typing import Optional
+
+import demoji
+from loguru import logger
+
+from ai_plays_jackbox.bot.jackbox5.bot_base import JackBox5BotBase
+
+_WORD_PROMPT_TEMPLATE = """
+You are playing Mad Verse City. You are being asked to come up with a word.
+
+{prompt}
+
+When generating your response, follow these rules:
+- You response must be {max_length} characters or less. It must be a singular word
+- Your personality is: {personality}
+- Do not include quotes in your response or any newlines, just the response itself.
+- Some suggestions for the word are {suggestions}
+"""
+
+_RHYME_PROMPT_TEMPLATE = """
+You are playing Mad Verse City.
+
+{prompt}
+
+When generating your response, follow these rules:
+- Your response must rhyme with {rhyme_word}. It cannot be the same word as that.
+- You response must be {max_length} characters or less.
+- Your personality is: {personality}
+- Do not include quotes in your response or any newlines, just the response itself.
+"""
+
+
+class MadVerseCityBot(JackBox5BotBase):
+    _actual_player_operation_key: Optional[str] = None
+    _current_word: str
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+    def _handle_welcome(self, data: dict):
+        pass
+
+    def _handle_player_operation(self, data: dict):
+        if not data:
+            return
+        room_state = data.get("state", None)
+        if not room_state:
+            return
+
+        prompt: dict[str, str] = data.get("prompt", {})
+        prompt_html = prompt.get("html", "")
+        clean_prompt = self._html_to_text(prompt_html)
+
+        max_length = data.get("maxLength", 40)
+        suggestions = data.get("suggestions", [])
+        if not suggestions:
+            suggestions = []
+        choices: list[dict] = data.get("choices", [])
+
+        match room_state:
+            case "EnterSingleText":
+                if "Give me a" in clean_prompt:
+                    word = self._generate_word(clean_prompt, max_length - 10, suggestions)
+                    self._current_word = demoji.replace(word, "")
+                    self._client_send({"action": "write", "entry": word})
+
+                if "Now, write a line to rhyme with" in clean_prompt:
+                    rhyme = self._generate_rhyme(clean_prompt, max_length - 10, self._current_word)
+                    rhyme = demoji.replace(rhyme, "")
+                    self._client_send({"action": "write", "entry": rhyme})
+
+            case "MakeSingleChoice":
+                if not choices:
+                    pass
+                if "Who won this battle" in data.get("prompt", {}).get("text", "") and data.get("chosen", None) != 0:
+                    choice_indexes = [i for i in range(0, len(choices))]
+                    choice = random.choice(choice_indexes)
+                    self._client_send({"action": "choose", "choice": choice})
+
+                if (
+                    "Press this button to skip the tutorial" in data.get("prompt", {}).get("text", "")
+                    and data.get("chosen", None) != 0
+                ):
+                    self._client_send({"action": "choose", "choice": 0})
+
+    def _handle_room_operation(self, data: dict):
+        pass
+
+    def _generate_word(self, prompt: str, max_length: int, suggestions: list[str]) -> str:
+        logger.info("Generating word...")
+        formatted_prompt = _WORD_PROMPT_TEMPLATE.format(
+            personality=self._personality,
+            prompt=prompt,
+            max_length=max_length,
+            suggestions=", ".join(suggestions),
+        )
+
+        word = self._chat_model.generate_text(
+            formatted_prompt,
+            "",
+            temperature=self._chat_model._chat_model_temperature,
+            top_p=self._chat_model._chat_model_top_p,
+        )
+        return word
+
+    def _generate_rhyme(self, prompt: str, max_length: int, rhyme_word: str) -> str:
+        logger.info("Generating rhyme...")
+        formatted_prompt = _RHYME_PROMPT_TEMPLATE.format(
+            personality=self._personality,
+            prompt=prompt,
+            max_length=max_length,
+            rhyme_word=rhyme_word,
+        )
+        rhyme = self._chat_model.generate_text(
+            formatted_prompt,
+            "",
+            temperature=self._chat_model._chat_model_temperature,
+            top_p=self._chat_model._chat_model_top_p,
+        )
+        return rhyme