mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-10 03:28:53 +00:00
fixes test
This commit is contained in:
parent
d32890ba55
commit
b0a2e08a60
1 changed files with 5 additions and 1 deletions
|
|
@ -2,12 +2,16 @@ from datetime import datetime, timedelta
|
|||
from unittest.mock import AsyncMock, patch
|
||||
|
||||
import pytest
|
||||
from fastapi import FastAPI
|
||||
from fastapi.testclient import TestClient
|
||||
from litellm_enterprise.proxy.audit_logging_endpoints import router as audit_router
|
||||
from litellm_enterprise.types.proxy.audit_logging_endpoints import AuditLogResponse
|
||||
|
||||
from litellm.proxy._types import UserAPIKeyAuth
|
||||
from litellm.proxy.proxy_server import app
|
||||
|
||||
# Create an app with just the audit router for testing
|
||||
app = FastAPI()
|
||||
app.include_router(audit_router)
|
||||
client = TestClient(app)
|
||||
|
||||
# Mock data for testing
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue