fix ci/cd pipeline

This commit is contained in:
Ishaan Jaff 2024-08-13 20:33:33 -07:00
parent 86818ddffc
commit a37f087b62
2 changed files with 3 additions and 1 deletions

View file

@ -125,6 +125,7 @@ jobs:
pip install tiktoken
pip install aiohttp
pip install click
pip install "boto3==1.34.34"
pip install jinja2
pip install tokenizers
pip install openai
@ -287,6 +288,7 @@ jobs:
pip install "pytest==7.3.1"
pip install "pytest-mock==3.12.0"
pip install "pytest-asyncio==0.21.1"
pip install "boto3==1.34.34"
pip install mypy
pip install pyarrow
pip install numpydoc

View file

@ -23,7 +23,7 @@ from litellm import RateLimitError, Timeout, completion, completion_cost, embedd
from litellm.llms.custom_httpx.http_handler import AsyncHTTPHandler, HTTPHandler
from litellm.llms.prompt_templates.factory import anthropic_messages_pt
# litellm.num_retries = 3
# litellm.num_retries =3
litellm.cache = None
litellm.success_callback = []
user_message = "Write a short poem about the sky"