gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

from https://github.com/BerriAI/litellm/issues/16810 LITELLM_LOCAL_MODEL_COST_MAP: True

Browse source
This commit is contained in:
Pavel Zorin 2025-11-19 19:23:37 +01:00
parent 5da997541b
commit 7b91f2182b
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
.github/workflows/e2e_tests.yml vendored
View file

@ -374,6 +374,8 @@ jobs:
- name: Run Entity Extraction Test - name: Run Entity Extraction Test
env: env:
ENV: 'dev' ENV: 'dev'
# remove when https://github.com/BerriAI/litellm/issues/16810 is fixed
LITELLM_LOCAL_MODEL_COST_MAP: "True"
LLM_MODEL: ${{ secrets.LLM_MODEL }} LLM_MODEL: ${{ secrets.LLM_MODEL }}
LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }} LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
LLM_API_KEY: ${{ secrets.LLM_API_KEY }} LLM_API_KEY: ${{ secrets.LLM_API_KEY }}
@ -402,6 +404,8 @@ jobs:
- name: Run Feedback Enrichment Test - name: Run Feedback Enrichment Test
env: env:
ENV: 'dev' ENV: 'dev'
# remove when https://github.com/BerriAI/litellm/issues/16810 is fixed
LITELLM_LOCAL_MODEL_COST_MAP: "True"
LLM_MODEL: ${{ secrets.LLM_MODEL }} LLM_MODEL: ${{ secrets.LLM_MODEL }}
LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }} LLM_ENDPOINT: ${{ secrets.LLM_ENDPOINT }}
LLM_API_KEY: ${{ secrets.LLM_API_KEY }} LLM_API_KEY: ${{ secrets.LLM_API_KEY }}

4
cognee/tests/test_cognee_server_start.py
View file

@ -9,9 +9,7 @@ from pathlib import Path
import sys import sys
import uuid import uuid
from tenacity import retry, stop_after_attempt, wait_fixed, wait_incrementing from tenacity import retry, stop_after_attempt, wait_incrementing
from cognee.api.health import HealthStatus
class TestCogneeServerStart(unittest.TestCase): class TestCogneeServerStart(unittest.TestCase):