mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-05 12:23:28 +00:00
test_proxy_stream
This commit is contained in:
@@ -1,21 +1,21 @@
|
||||
import openai
|
||||
import os
|
||||
# import openai
|
||||
# import os
|
||||
|
||||
os.environ["OPENAI_API_KEY"] = ""
|
||||
# os.environ["OPENAI_API_KEY"] = ""
|
||||
|
||||
openai.api_key = os.environ["OPENAI_API_KEY"]
|
||||
openai.api_base ="http://localhost:5000"
|
||||
# openai.api_key = os.environ["OPENAI_API_KEY"]
|
||||
# openai.api_base ="http://localhost:5000"
|
||||
|
||||
messages = [
|
||||
{
|
||||
"role": "user",
|
||||
"content": "write a 1 pg essay in liteLLM"
|
||||
}
|
||||
]
|
||||
# messages = [
|
||||
# {
|
||||
# "role": "user",
|
||||
# "content": "write a 1 pg essay in liteLLM"
|
||||
# }
|
||||
# ]
|
||||
|
||||
response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages, stream=True)
|
||||
print("got response", response)
|
||||
# response is a generator
|
||||
# response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages, stream=True)
|
||||
# print("got response", response)
|
||||
# # response is a generator
|
||||
|
||||
for chunk in response:
|
||||
print(chunk)
|
||||
# for chunk in response:
|
||||
# print(chunk)
|
||||
|
||||
+1
-1
@@ -1,6 +1,6 @@
|
||||
[tool.poetry]
|
||||
name = "litellm"
|
||||
version = "0.1.406"
|
||||
version = "0.1.408"
|
||||
description = "Library to easily interface with LLM API providers"
|
||||
authors = ["BerriAI"]
|
||||
license = "MIT License"
|
||||
|
||||
Reference in New Issue
Block a user