tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-10 22:24:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

test parallelism on circle ci

Browse Source
This commit is contained in:
Krrish Dholakia
2023-09-05 14:47:50 -07:00
parent 343ff81f27
commit 9a82bfa4d2
2 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_completion.py
+12 -12
View File
@@ -410,18 +410,18 @@ def test_completion_together_ai():
pytest.fail(f"Error occurred: {e}")
def test_customprompt_together_ai():
try:
litellm.register_prompt_template(
model="OpenAssistant/llama2-70b-oasst-sft-v10",
roles={"system":"<|im_start|>system", "assistant":"<|im_start|>assistant", "user":"<|im_start|>user"}, # tell LiteLLM how you want to map the openai messages to this model
pre_message_sep= "\n",
post_message_sep= "\n"
)
response = completion(model="together_ai/OpenAssistant/llama2-70b-oasst-sft-v10", messages=messages)
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# def test_customprompt_together_ai():
# try:
# litellm.register_prompt_template(
# model="OpenAssistant/llama2-70b-oasst-sft-v10",
# roles={"system":"<|im_start|>system", "assistant":"<|im_start|>assistant", "user":"<|im_start|>user"}, # tell LiteLLM how you want to map the openai messages to this model
# pre_message_sep= "\n",
# post_message_sep= "\n"
# )
# response = completion(model="together_ai/OpenAssistant/llama2-70b-oasst-sft-v10", messages=messages)
# print(response)
# except Exception as e:
# pytest.fail(f"Error occurred: {e}")
def test_completion_sagemaker():
try:
pyproject.toml
+1 -1
View File
@@ -1,6 +1,6 @@
[tool.poetry]
name = "litellm"
version = "0.1.533"
version = "0.1.534"
description = "Library to easily interface with LLM API providers"
authors = ["BerriAI"]
license = "MIT License"