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

@ -249,7 +249,7 @@ client = OpenAI(
response = client.chat.completions.create( response = client.chat.completions.create(
# we select model from arch_config file # we select model from arch_config file
model="--", model="None",
messages=[{"role": "user", "content": "What is the capital of France?"}], messages=[{"role": "user", "content": "What is the capital of France?"}],
) )

View file

@ -190,7 +190,7 @@ impl HttpContext for StreamContext {
messages: deserialized_body.messages.clone(), messages: deserialized_body.messages.clone(),
metadata, metadata,
stream: deserialized_body.stream, stream: deserialized_body.stream,
model: "--".to_string(), model: deserialized_body.model.clone(),
stream_options: deserialized_body.stream_options.clone(), stream_options: deserialized_body.stream_options.clone(),
tools: Some(tool_calls), tools: Some(tool_calls),
}; };

View file

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

View file

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

View file

@ -238,7 +238,7 @@ POST {{model_server_endpoint}}/function_calling HTTP/1.1
Content-Type: application/json Content-Type: application/json
{ {
"model": "--", "model": "None",
"messages": [ "messages": [
{ {
"role": "user", "role": "user",

View file

@ -82,7 +82,7 @@ POST {{prompt_endpoint}}/v1/chat/completions HTTP/1.1
Content-Type: application/json Content-Type: application/json
{ {
"model": "--", "model": "None",
"messages": [ "messages": [
{ {
"role": "user", "role": "user",