Fix: Prediction model

immerSIR · immerSIR · commit db5de3527352 · 2026-02-23T04:12:02.000-05:00
diff --git a/app/services/cnn/openai_vision.py b/app/services/cnn/openai_vision.py
@@ -108,7 +108,7 @@ def predict(image_bytes) -> Tuple[List[Tuple[str, float]], List[float]]:
             "temperature": 1,
             "max_output_tokens": 2048,
             "top_p": 1,
-            "text": {"format": {"type": "json"}},
+            "text": {"format": {"type": "json_object"}},
             "reasoning": {},
             "store": True
         }
diff --git a/test/test_cnn/test_openai_vision.py b/test/test_cnn/test_openai_vision.py
@@ -73,7 +73,7 @@ def test_predict_with_successful_response(mock_openai_client, mock_image_bytes,
     assert call_args['temperature'] == 1
     assert call_args['max_output_tokens'] == 2048
     assert call_args['top_p'] == 1
-    assert call_args['text'] == {"format": {"type": "json"}}
+    assert call_args['text'] == {"format": {"type": "json_object"}}
     assert call_args['reasoning'] == {}
     assert call_args['store'] is True
 

Original file line number	Diff line number	Diff line change
`@@ -108,7 +108,7 @@ def predict(image_bytes) -> Tuple[List[Tuple[str, float]], List[float]]:`
`108`	`108`	`"temperature": 1,`
`109`	`109`	`"max_output_tokens": 2048,`
`110`	`110`	`"top_p": 1,`
`111`		`- "text": {"format": {"type": "json"}},`
	`111`	`+ "text": {"format": {"type": "json_object"}},`
`112`	`112`	`"reasoning": {},`
`113`	`113`	`"store": True`
`114`	`114`	`}`