From 4d82c815313758a112817c23c215018924043c2a Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Thu, 2 Nov 2023 20:53:39 -0700 Subject: [PATCH] (fix) proxy cli tests --- litellm/proxy/proxy_cli.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/litellm/proxy/proxy_cli.py b/litellm/proxy/proxy_cli.py index 59b90348773..d65adce16c6 100644 --- a/litellm/proxy/proxy_cli.py +++ b/litellm/proxy/proxy_cli.py @@ -176,13 +176,13 @@ def run_server(host, port, api_base, api_version, model, alias, add_key, headers openai.api_key = "temp-key" print(openai.api_base) - response = openai.Completion.create(model="gpt-3.5-turbo", prompt='hello who are you') + response = openai.Completion.create(model="gpt-3.5-turbo", prompt='this is a test request, write a short poem') print(response) response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages = [ { "role": "user", - "content": "this is a test request, acknowledge that you got it" + "content": "this is a test request, write a short poem" } ]) click.echo(f'LiteLLM: response from proxy {response}') @@ -191,7 +191,7 @@ def run_server(host, port, api_base, api_version, model, alias, add_key, headers response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages = [ { "role": "user", - "content": "this is a test request, acknowledge that you got it" + "content": "this is a test request, write a short poem" } ], stream=True,