phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 03:04:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test: remove local test

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-05 12:45:52 -08:00
parent a9b50a12c5
commit 397eefabe1
1 changed files with 30 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

59
litellm/proxy/tests/test_langchain_request.py
View file

@ -1,38 +1,39 @@
from langchain.chat_models import ChatOpenAI ## LOCAL TEST
from langchain.prompts.chat import ( # from langchain.chat_models import ChatOpenAI
ChatPromptTemplate, # from langchain.prompts.chat import (
HumanMessagePromptTemplate, # ChatPromptTemplate,
SystemMessagePromptTemplate, # HumanMessagePromptTemplate,
) # SystemMessagePromptTemplate,
from langchain.schema import HumanMessage, SystemMessage # )
# from langchain.schema import HumanMessage, SystemMessage
chat = ChatOpenAI( # chat = ChatOpenAI(
openai_api_base="http://0.0.0.0:8000", # openai_api_base="http://0.0.0.0:8000",
model = "gpt-3.5-turbo", # model = "gpt-3.5-turbo",
temperature=0.1 # temperature=0.1
) # )
messages = [ # messages = [
SystemMessage( # SystemMessage(
content="You are a helpful assistant that im using to make a test request to." # content="You are a helpful assistant that im using to make a test request to."
), # ),
HumanMessage( # HumanMessage(
content="test from litellm. tell me why it's amazing in 1 sentence" # content="test from litellm. tell me why it's amazing in 1 sentence"
), # ),
] # ]
response = chat(messages) # response = chat(messages)
print(response) # print(response)
claude_chat = ChatOpenAI( # claude_chat = ChatOpenAI(
openai_api_base="http://0.0.0.0:8000", # openai_api_base="http://0.0.0.0:8000",
model = "claude-v1", # model = "claude-v1",
temperature=0.1 # temperature=0.1
) # )
response = claude_chat(messages) # response = claude_chat(messages)
print(response) # print(response)