Changing model_validate to refusal checking

Author: pamelafox

basic_azure.py

@@ -36,7 +36,9 @@ class CalendarEvent(BaseModel):
     response_format=CalendarEvent,
 )
 
-output = completion.choices[0].message.parsed
-event = CalendarEvent.model_validate(output)
+message = completion.choices[0].message
+if (message.refusal):
+    rich.print(message.refusal)
+else:
+    rich.print(message.parsed)
 
-rich.print(event)

basic_githubmodels.py

@@ -28,7 +28,8 @@ class CalendarEvent(BaseModel):
     response_format=CalendarEvent,
 )
 
-output = completion.choices[0].message.parsed
-event = CalendarEvent.model_validate(output)
-
-rich.print(event)
+message = completion.choices[0].message
+if (message.refusal):
+    rich.print(message.refusal)
+else:
+    rich.print(message.parsed)

extract_github_issue.py

@@ -79,6 +79,9 @@ class HackSubmission(BaseModel):
     response_format=HackSubmission,
 )
 
-output = completion.choices[0].message.parsed
-hack_submission = HackSubmission.model_validate(output)
-print(hack_submission)
+message = completion.choices[0].message
+if (message.refusal):
+    print(message.refusal)
+else:
+    print(message.parsed)
+

extract_github_repo.py

@@ -6,7 +6,7 @@
 import azure.identity
 import openai
 import requests
-import rich
+from rich import print
 from dotenv import load_dotenv
 from pydantic import BaseModel, Field
 
@@ -94,6 +94,8 @@ class RepoOverview(BaseModel):
     response_format=RepoOverview,
 )
 
-output = completion.choices[0].message.parsed
-repo_overview = RepoOverview.model_validate(output)
-rich.print(repo_overview)
+message = completion.choices[0].message
+if (message.refusal):
+    print(message.refusal)
+else:
+    print(message.parsed)

extract_image_graph.py

@@ -72,6 +72,9 @@ def open_image_as_base64(filename):
     response_format=Graph,
 )
 
-output = completion.choices[0].message.parsed
-graph = Graph.model_validate(output)
-print(graph)
+message = completion.choices[0].message
+if (message.refusal):
+    print(message.refusal)
+else:
+    print(message.parsed)
+

extract_image_table.py

@@ -80,6 +80,8 @@ def open_image_as_base64(filename):
     response_format=PlantInventory,
 )
 
-output = completion.choices[0].message.parsed
-plant_inventory = PlantInventory.model_validate(output)
-print(plant_inventory)
+message = completion.choices[0].message
+if (message.refusal):
+    print(message.refusal)
+else:
+    print(message.parsed)

extract_pdf_receipt.py

@@ -66,6 +66,9 @@ class Receipt(BaseModel):
     response_format=Receipt,
 )
 
-output = completion.choices[0].message.parsed
-receipt = Receipt.model_validate(output)
-print(receipt)
+message = completion.choices[0].message
+if (message.refusal):
+    print(message.refusal)
+else:
+    print(message.parsed)
+

extract_webpage.py

@@ -67,6 +67,9 @@ class BlogPost(BaseModel):
     response_format=BlogPost,
 )
 
-output = completion.choices[0].message.parsed
-blog_post = BlogPost.model_validate(output)
-print(blog_post)
+message = completion.choices[0].message
+if (message.refusal):
+    print(message.refusal)
+else:
+    print(message.parsed)
+