refactor: improve readability in trajectories.py and server.py

- Simplified the construction of loggable_dict in the Trajectory class for better clarity.
- Enhanced assertion error message formatting in OpenAICompatibleTinkerServer for consistency.

Author: bradhilton

src/art/tinker/server.py

@@ -133,9 +133,9 @@ async def chat_completions(
             choices: list[Choice] = []
             for i, sequence in enumerate(sample_response.sequences):
                 assert sequence.logprobs is not None, "Logprobs are required"
-                assert len(sequence.tokens) == len(
-                    sequence.logprobs
-                ), "Tokens and logprobs must have the same length"
+                assert len(sequence.tokens) == len(sequence.logprobs), (
+                    "Tokens and logprobs must have the same length"
+                )
                 rendered_response_tokens = renderer.tokenizer.encode(
                     renderer.tokenizer.decode(sequence.tokens)
                 )

src/art/trajectories.py

@@ -92,13 +92,9 @@ def for_logging(self) -> dict[str, Any]:
         for message_or_choice in self.messages_and_choices:
             trainable = isinstance(message_or_choice, Choice)
             message = (
-                message_or_choice.message.to_dict()
-                if trainable
-                else message_or_choice  # ty:ignore[possibly-missing-attribute]
+                message_or_choice.message.to_dict() if trainable else message_or_choice  # ty:ignore[possibly-missing-attribute]
             )
-            loggable_dict["messages"].append(
-                {**message, "trainable": trainable}
-            )  # ty:ignore[invalid-argument-type, possibly-missing-attribute]
+            loggable_dict["messages"].append({**message, "trainable": trainable})  # ty:ignore[invalid-argument-type, possibly-missing-attribute]
         return loggable_dict