Skip to content

Commit 77fe863

Browse files
authored
feat: fix litellm version (#1414)
1 parent 4684609 commit 77fe863

File tree

4 files changed

+37
-14
lines changed

4 files changed

+37
-14
lines changed

aperag/llm/litellm_logging.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ def setup_litellm_logging():
2424
import litellm
2525

2626
litellm.suppress_debug_info = True
27+
litellm.disable_streaming_logging = True
2728
logging.info("LiteLLM debug info suppression is enabled.")
2829

2930
# Filter Pydantic serialization warnings globally

aperag/llm/litellm_track.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -77,12 +77,3 @@ def log_post_api_call_for_completion(self, kwargs, response_obj, start_time, end
7777

7878
def register_custom_llm_track():
7979
litellm.callbacks = [MyCustomHandler()]
80-
81-
82-
def register_opik_llm_track():
83-
from litellm.integrations.opik.opik import OpikLogger
84-
85-
from aperag.config import settings
86-
87-
if settings.opik_api_key and settings.opik_workspace:
88-
litellm.callbacks = [OpikLogger()]

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ dependencies = [
4242
"numpy>=2.1.3",
4343
"pipmaster<1.0.0,>=0.5.4",
4444
"pytablewriter<2.0.0,>=1.2.1",
45-
"litellm>=1.76.0",
45+
"litellm>=v1.80.0",
4646
"requests",
4747
"asyncpg>=0.30.0",
4848
"markitdown[all]>=0.1.1",

uv.lock

Lines changed: 35 additions & 4 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)