From 3761f46e60c59ac7ccb770aedcf339bbf2a26e90 Mon Sep 17 00:00:00 2001 From: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Thu, 19 Dec 2024 02:37:48 +0000 Subject: [PATCH] style: fix JSON formatting in test_rest_api.py per ruff-format Co-Authored-By: Alex Reibman --- tests/test_rest_api.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/tests/test_rest_api.py b/tests/test_rest_api.py index cfb2e070..db5c1b59 100644 --- a/tests/test_rest_api.py +++ b/tests/test_rest_api.py @@ -75,14 +75,11 @@ def test_create_events(self) -> bool: "model": "gpt-4", "prompt": [ {"role": "system", "content": "You are a helpful assistant"}, - {"role": "user", "content": "Analyze this data..."} + {"role": "user", "content": "Analyze this data..."}, ], - "completion": { - "role": "assistant", - "content": "Based on the data..." - }, + "completion": {"role": "assistant", "content": "Based on the data..."}, "prompt_tokens": 150, - "completion_tokens": 80 + "completion_tokens": 80, }, { "id": str(uuid.uuid4()), @@ -92,8 +89,8 @@ def test_create_events(self) -> bool: "end_timestamp": now, "session_id": self.session_id, "input": "SELECT * FROM users", - "output": "Retrieved 5 users" - } + "output": "Retrieved 5 users", + }, ] }