mirror of
https://github.com/kennethreitz/instructor.git
synced 2026-06-05 22:50:18 +00:00
Jason Liu
41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
from typing_extensions import Annotated
|
|
from pydantic import BaseModel, ValidationError, AfterValidator
|
|
from openai import OpenAI
|
|
|
|
import instructor
|
|
|
|
client = instructor.patch(OpenAI())
|
|
|
|
|
|
def no_competitors(v: str) -> str:
|
|
# does not allow the competitors of mcdonalds
|
|
competitors = ["burger king", "wendy's", "carl's jr", "jack in the box"]
|
|
for competitor in competitors:
|
|
if competitor in v.lower():
|
|
raise ValueError(
|
|
f"""Let them know that you are work for and are only allowed to talk about mcdonalds.
|
|
Do not apologize. Do not even mention `{competitor}` since they are a a competitor of McDonalds"""
|
|
)
|
|
return v
|
|
|
|
|
|
class Response(BaseModel):
|
|
message: Annotated[str, AfterValidator(no_competitors)]
|
|
|
|
|
|
try:
|
|
resp = client.chat.completions.create(
|
|
model="gpt-3.5-turbo",
|
|
response_model=Response,
|
|
max_retries=2,
|
|
messages=[
|
|
{
|
|
"role": "user",
|
|
"content": "What is your favourite order at burger king?",
|
|
},
|
|
],
|
|
) # type: ignore
|
|
print(resp.model_dump_json(indent=2))
|
|
except ValidationError as e:
|
|
print(e)
|