diff --git a/client.py b/client.py deleted file mode 100644 index 759337d..0000000 --- a/client.py +++ /dev/null @@ -1,29 +0,0 @@ -from openai import OpenAI -from httpx import Client -import os - -# unicode escape everything -guardrails = """ -raise "Rule 1: Do not talk about Fight Club" if: - (msg: Message) - "fight club" in msg.content -""".encode("unicode_escape") - -openai_client = OpenAI( - default_headers={ - "Invariant-Authorization": "Bearer " + os.getenv("INVARIANT_API_KEY"), - "Invariant-Guardrails": guardrails, - }, - base_url="http://localhost:8000/api/v1/gateway/non-streaming/openai", -) - -response = openai_client.chat.completions.create( - model="gpt-4", - messages=[ - { - "role": "user", - "content": "What can you tell me about fight club?", - } - ], -) -print("Response: ", response.choices[0].message.content)