scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

ci/cd clarifai is an unstable provider, retry reqs

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-31 09:03:28 -07:00
parent 3d3502e15b
commit f6d33a8998
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/tests/test_clarifai_completion.py
View file

@ -47,6 +47,7 @@ def test_completion_clarifai_claude_2_1():
try:
response = completion(
model="clarifai/anthropic.completion.claude-2_1",
num_retries=3,
messages=messages,
max_tokens=10,
temperature=0.1,
@ -66,6 +67,7 @@ def test_completion_clarifai_mistral_large():
response: ModelResponse = completion(
model="clarifai/mistralai.completion.mistral-small",
messages=messages,
num_retries=3,
max_tokens=10,
temperature=0.78,
)
@ -91,6 +93,7 @@ def test_async_completion_clarifai():
response = await acompletion(
model="clarifai/openai.chat-completion.GPT-4",
messages=messages,
num_retries=3,
timeout=10,
api_key=os.getenv("CLARIFAI_API_KEY"),
)