From 6377b1912b58d92438311cfc9547c1bb2b6de9b3 Mon Sep 17 00:00:00 2001
From: Neil Mehta <neil@lmstudio.ai>
Date: Mon, 24 Mar 2025 18:54:52 -0400
Subject: [PATCH] Revert "Use int for year in test case"

This reverts commit 5426341d5be7e00d95a933c0cae24715bd2436d2.
---
 tests/integration/inference/test_text_inference.py     | 4 ++--
 tests/integration/test_cases/inference/completion.json | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/integration/inference/test_text_inference.py b/tests/integration/inference/test_text_inference.py
index 5410f20c5..a3cfce4fd 100644
--- a/tests/integration/inference/test_text_inference.py
+++ b/tests/integration/inference/test_text_inference.py
@@ -199,8 +199,8 @@ def test_text_completion_structured_output(client_with_models, text_model_id, te
 
     class AnswerFormat(BaseModel):
         name: str
-        year_born: int
-        year_retired: int
+        year_born: str
+        year_retired: str
 
     tc = TestCase(test_case)
 
diff --git a/tests/integration/test_cases/inference/completion.json b/tests/integration/test_cases/inference/completion.json
index 447087d19..731ceddbc 100644
--- a/tests/integration/test_cases/inference/completion.json
+++ b/tests/integration/test_cases/inference/completion.json
@@ -40,8 +40,8 @@
             "user_input": "Michael Jordan was born in 1963. He played basketball for the Chicago Bulls. He retired in 2003.",
             "expected": {
                 "name": "Michael Jordan",
-                "year_born": 1963,
-                "year_retired": 2003
+                "year_born": "1963",
+                "year_retired": "2003"
             }
         }
     },