fix tests

This commit is contained in:
Adil Hafeez 2025-03-21 15:02:36 -07:00
parent ff2ddafab7
commit 678963a5ea
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
6 changed files with 7 additions and 7 deletions

View file

@ -38,7 +38,7 @@ def chat(
try:
response = client.chat.completions.create(
# we select model from arch_config file
model="--",
model="None",
messages=history,
temperature=1.0,
stream=True,

View file

@ -54,13 +54,13 @@ def chat(
if model_selector and model_selector != "":
headers["x-arch-llm-provider-hint"] = model_selector
client = OpenAI(
api_key="--",
api_key="None",
base_url=CHAT_COMPLETION_ENDPOINT,
default_headers=headers,
)
response = client.chat.completions.create(
# we select model from arch_config file
model="--",
model="None",
messages=history,
temperature=1.0,
stream=True,