ai-plays-jackbox 0.2.1__py3-none-any.whl → 0.2.2__py3-none-any.whl

ai_plays_jackbox/bot/bot_base.py

@@ -2,7 +2,7 @@ import json
 import threading
 import traceback
 from abc import ABC, abstractmethod
-from typing import Optional
+from typing import Optional, Union
 from urllib import parse
 from uuid import uuid4
 
@@ -110,18 +110,17 @@ class JackBoxBotBase(ABC):
     def _on_message(self, wsapp, message) -> None:
         server_message = ServerMessage.model_validate_json(message)
 
-        # logger.info(server_message.result)
-
         if server_message.opcode == "client/welcome":
             self._player_id = server_message.result["id"]
             self._handle_welcome(server_message.result)
 
-        elif server_message.opcode == "object" or server_message.opcode == "text":
-            if server_message.opcode == "object":
-                operation = ObjectOperation(**server_message.result)
-            elif server_message.opcode == "text":
-                operation = TextOperation(**server_message.result)  # type: ignore
+        operation: Optional[Union[ObjectOperation, TextOperation]] = None
+        if server_message.opcode == "object":
+            operation = ObjectOperation(**server_message.result)
+        elif server_message.opcode == "text":
+            operation = TextOperation(**server_message.result)
 
+        if operation is not None:
             if self._is_player_operation_key(operation.key):
                 self._handle_player_operation(operation.json_data)
             if self._is_room_operation_key(operation.key):

ai_plays_jackbox/llm/gemini_model.py

@@ -22,7 +22,7 @@ class GeminiModel(ChatModel):
 
         # Check connection and if model exists, this will hard fail if connection can't be made
         # Or if the model is not found
-        _ = self._gemini_vertex_ai_client.models.get(self._model)
+        _ = self._gemini_vertex_ai_client.models.get(model=self._model)
 
     @classmethod
     def get_default_model(cls):

{ai_plays_jackbox-0.2.1.dist-info → ai_plays_jackbox-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: ai-plays-jackbox
-Version: 0.2.1
+Version: 0.2.2
 Summary: Bringing the dead internet theory to life. Have AI play JackBox with you; no friends required!
 License: MIT
 Author: Daniel S. Thompson
@@ -28,7 +28,7 @@ Project-URL: Bug Tracker, https://github.com/SudoSpartanDan/AIPlaysJackBox/issue
 Project-URL: Repository, https://github.com/SudoSpartanDan/AIPlaysJackBox
 Description-Content-Type: text/markdown
 
-# AI Plays JackBox
+# 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)

{ai_plays_jackbox-0.2.1.dist-info → ai_plays_jackbox-0.2.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 ai_plays_jackbox/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_plays_jackbox/bot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ai_plays_jackbox/bot/bot_base.py,sha256=d9PVr2fA6ajeTs1tU-Qa1ZOMuOu-hQOEHLiHHpzbp1A,7415
+ai_plays_jackbox/bot/bot_base.py,sha256=c9luRy5GPsjZdL87GU5T2kj4NqroNU3DyzX5RXCgpg0,7369
 ai_plays_jackbox/bot/bot_factory.py,sha256=7FuDvYxotttPGEW5iHL-52nrfTn-hFJGKVwwsxkOak8,1264
 ai_plays_jackbox/bot/bot_personality.py,sha256=x7oVyyH83Ofd8C7ZHczEVgY_6NSpSafixZCI-NKxBc8,4217
 ai_plays_jackbox/bot/jackbox5/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -25,7 +25,7 @@ ai_plays_jackbox/constants.py,sha256=5ejveTgb9_QJOqYW8Lc9mqrUIFo6LjE-GJYJEGHpSM0
 ai_plays_jackbox/llm/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 ai_plays_jackbox/llm/chat_model.py,sha256=VirseP5NgCVg42-NGO5qj_v4bXPqMl1c98aUX8pRHOY,984
 ai_plays_jackbox/llm/chat_model_factory.py,sha256=rJNMXyOIDf6CnX41EWt3mW7u8m30uRhq_I8dXsI4sf0,1222
-ai_plays_jackbox/llm/gemini_model.py,sha256=Zq9WGRCevetb9y7dJrpr4Wn0XCbY6aN5N9q0lI_Qna0,2879
+ai_plays_jackbox/llm/gemini_model.py,sha256=ElcfHC-r_fOYGEH7CEthisj8FCLGhe6-QLh-RNwZjN4,2885
 ai_plays_jackbox/llm/ollama_model.py,sha256=DDc8pRdpXVDnq3k1v1VsJUAaA-3HaFefEIimmptsU5I,1644
 ai_plays_jackbox/llm/openai_model.py,sha256=pDW6-Afy6dQh84ccM8lTZjz5bsymtaxzTgW8R_2BMtE,2734
 ai_plays_jackbox/room/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -35,8 +35,8 @@ ai_plays_jackbox/scripts/lint.py,sha256=xCQ9vYZv0iiRvaFgdoWr9Gpe2Tz9rX9fn1ltZFm3
 ai_plays_jackbox/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 ai_plays_jackbox/ui/main.py,sha256=-J7DGanJrMkPUP0OdVPVP_IIr5dZ6G2D1IehV365bGo,242
 ai_plays_jackbox/ui/startup.py,sha256=s_Hfp90NZ73egs_ePfWk0rEUYoctFXPU5jqsKbtXY8I,9057
-ai_plays_jackbox-0.2.1.dist-info/LICENSE,sha256=Dxj3CeEsn2Olpkn5QJxGmQemd3zGnM2aA-7Hme4uIE4,1095
-ai_plays_jackbox-0.2.1.dist-info/METADATA,sha256=DKyAqsyLiTQh2JMrGbHKsnrsq4kPFUrYpokTw3VIXMk,7605
-ai_plays_jackbox-0.2.1.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-ai_plays_jackbox-0.2.1.dist-info/entry_points.txt,sha256=FtRnZSjr8xKAC206XNhh-M7vJlcF7Hb_tBg4qZFzQvY,117
-ai_plays_jackbox-0.2.1.dist-info/RECORD,,
+ai_plays_jackbox-0.2.2.dist-info/LICENSE,sha256=Dxj3CeEsn2Olpkn5QJxGmQemd3zGnM2aA-7Hme4uIE4,1095
+ai_plays_jackbox-0.2.2.dist-info/METADATA,sha256=DOgi4733M3qRZVjGg_lvZIx3YVVBh9ok1XU75DxjTRQ,7605
+ai_plays_jackbox-0.2.2.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+ai_plays_jackbox-0.2.2.dist-info/entry_points.txt,sha256=FtRnZSjr8xKAC206XNhh-M7vJlcF7Hb_tBg4qZFzQvY,117
+ai_plays_jackbox-0.2.2.dist-info/RECORD,,