fixing outline tests

adtyavrdhn · adtyavrdhn · commit cc58097ebd60 · 2025-11-08T17:48:28.000+05:30
diff --git a/tests/models/test_outlines.py b/tests/models/test_outlines.py
@@ -324,11 +324,9 @@ async def test_request_async(llamacpp_model: OutlinesModel) -> None:
                     )
                 ],
                 instructions='Answer in one word.',
+                run_id=IsStr(),
             ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
-            ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
     result = await agent.run('What is the capital of Germany?', message_history=result.all_messages())
@@ -342,11 +340,9 @@ async def test_request_async(llamacpp_model: OutlinesModel) -> None:
                     )
                 ],
                 instructions='Answer in one word.',
+                run_id=IsStr(),
             ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
-            ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
             ModelRequest(
                 parts=[
                     UserPromptPart(
@@ -355,11 +351,9 @@ async def test_request_async(llamacpp_model: OutlinesModel) -> None:
                     )
                 ],
                 instructions='Answer in one word.',
+                run_id=IsStr(),
             ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
-            ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
 
@@ -376,12 +370,10 @@ def test_request_sync(llamacpp_model: OutlinesModel) -> None:
                         content='What is the capital of France?',
                         timestamp=IsDatetime(),
                     )
-                ]
-            ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
+                ],
+                run_id=IsStr(),
             ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
 
@@ -408,12 +400,10 @@ async def test_request_async_model(mock_async_model: OutlinesModel) -> None:
                         content='What is the capital of France?',
                         timestamp=IsDatetime(),
                     )
-                ]
-            ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
+                ],
+                run_id=IsStr(),
             ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
 
@@ -445,12 +435,10 @@ def test_request_image_binary(transformers_multimodal_model: OutlinesModel, bina
                         ],
                         timestamp=IsDatetime(),
                     )
-                ]
-            ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
+                ],
+                run_id=IsStr(),
             ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
 
@@ -478,12 +466,10 @@ def test_request_image_url(transformers_multimodal_model: OutlinesModel) -> None
                         ],
                         timestamp=IsDatetime(),
                     )
-                ]
-            ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
+                ],
+                run_id=IsStr(),
             ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
 
@@ -536,12 +522,10 @@ class Box(BaseModel):
                         content='Give me the dimensions of a box',
                         timestamp=IsDatetime(),
                     )
-                ]
-            ),
-            ModelResponse(
-                parts=[TextPart(content=IsStr())],
-                timestamp=IsDatetime(),
+                ],
+                run_id=IsStr(),
             ),
+            ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),
         ]
     )
 

Original file line number	Diff line number	Diff line change
`@@ -324,11 +324,9 @@ async def test_request_async(llamacpp_model: OutlinesModel) -> None:`
`324`	`324`	`)`
`325`	`325`	`],`
`326`	`326`	`instructions='Answer in one word.',`
	`327`	`+ run_id=IsStr(),`
`327`	`328`	`),`
`328`		`- ModelResponse(`
`329`		`- parts=[TextPart(content=IsStr())],`
`330`		`- timestamp=IsDatetime(),`
`331`		`- ),`
	`329`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`332`	`330`	`]`
`333`	`331`	`)`
`334`	`332`	`result = await agent.run('What is the capital of Germany?', message_history=result.all_messages())`
`@@ -342,11 +340,9 @@ async def test_request_async(llamacpp_model: OutlinesModel) -> None:`
`342`	`340`	`)`
`343`	`341`	`],`
`344`	`342`	`instructions='Answer in one word.',`
	`343`	`+ run_id=IsStr(),`
`345`	`344`	`),`
`346`		`- ModelResponse(`
`347`		`- parts=[TextPart(content=IsStr())],`
`348`		`- timestamp=IsDatetime(),`
`349`		`- ),`
	`345`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`350`	`346`	`ModelRequest(`
`351`	`347`	`parts=[`
`352`	`348`	`UserPromptPart(`
`@@ -355,11 +351,9 @@ async def test_request_async(llamacpp_model: OutlinesModel) -> None:`
`355`	`351`	`)`
`356`	`352`	`],`
`357`	`353`	`instructions='Answer in one word.',`
	`354`	`+ run_id=IsStr(),`
`358`	`355`	`),`
`359`		`- ModelResponse(`
`360`		`- parts=[TextPart(content=IsStr())],`
`361`		`- timestamp=IsDatetime(),`
`362`		`- ),`
	`356`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`363`	`357`	`]`
`364`	`358`	`)`
`365`	`359`
`@@ -376,12 +370,10 @@ def test_request_sync(llamacpp_model: OutlinesModel) -> None:`
`376`	`370`	`content='What is the capital of France?',`
`377`	`371`	`timestamp=IsDatetime(),`
`378`	`372`	`)`
`379`		`- ]`
`380`		`- ),`
`381`		`- ModelResponse(`
`382`		`- parts=[TextPart(content=IsStr())],`
`383`		`- timestamp=IsDatetime(),`
	`373`	`+ ],`
	`374`	`+ run_id=IsStr(),`
`384`	`375`	`),`
	`376`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`385`	`377`	`]`
`386`	`378`	`)`
`387`	`379`
`@@ -408,12 +400,10 @@ async def test_request_async_model(mock_async_model: OutlinesModel) -> None:`
`408`	`400`	`content='What is the capital of France?',`
`409`	`401`	`timestamp=IsDatetime(),`
`410`	`402`	`)`
`411`		`- ]`
`412`		`- ),`
`413`		`- ModelResponse(`
`414`		`- parts=[TextPart(content=IsStr())],`
`415`		`- timestamp=IsDatetime(),`
	`403`	`+ ],`
	`404`	`+ run_id=IsStr(),`
`416`	`405`	`),`
	`406`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`417`	`407`	`]`
`418`	`408`	`)`
`419`	`409`
`@@ -445,12 +435,10 @@ def test_request_image_binary(transformers_multimodal_model: OutlinesModel, bina`
`445`	`435`	`],`
`446`	`436`	`timestamp=IsDatetime(),`
`447`	`437`	`)`
`448`		`- ]`
`449`		`- ),`
`450`		`- ModelResponse(`
`451`		`- parts=[TextPart(content=IsStr())],`
`452`		`- timestamp=IsDatetime(),`
	`438`	`+ ],`
	`439`	`+ run_id=IsStr(),`
`453`	`440`	`),`
	`441`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`454`	`442`	`]`
`455`	`443`	`)`
`456`	`444`
`@@ -478,12 +466,10 @@ def test_request_image_url(transformers_multimodal_model: OutlinesModel) -> None`
`478`	`466`	`],`
`479`	`467`	`timestamp=IsDatetime(),`
`480`	`468`	`)`
`481`		`- ]`
`482`		`- ),`
`483`		`- ModelResponse(`
`484`		`- parts=[TextPart(content=IsStr())],`
`485`		`- timestamp=IsDatetime(),`
	`469`	`+ ],`
	`470`	`+ run_id=IsStr(),`
`486`	`471`	`),`
	`472`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`487`	`473`	`]`
`488`	`474`	`)`
`489`	`475`
`@@ -536,12 +522,10 @@ class Box(BaseModel):`
`536`	`522`	`content='Give me the dimensions of a box',`
`537`	`523`	`timestamp=IsDatetime(),`
`538`	`524`	`)`
`539`		`- ]`
`540`		`- ),`
`541`		`- ModelResponse(`
`542`		`- parts=[TextPart(content=IsStr())],`
`543`		`- timestamp=IsDatetime(),`
	`525`	`+ ],`
	`526`	`+ run_id=IsStr(),`
`544`	`527`	`),`
	`528`	`+ ModelResponse(parts=[TextPart(content=IsStr())], timestamp=IsDatetime(), run_id=IsStr()),`
`545`	`529`	`]`
`546`	`530`	`)`
`547`	`531`