forked from phoenix/litellm-mirror
init gcs using gcs_bucket
This commit is contained in:
parent
1595dc4dc6
commit
6c899ced28
3 changed files with 15 additions and 1 deletions
|
@ -46,6 +46,7 @@ _custom_logger_compatible_callbacks_literal = Literal[
|
||||||
"galileo",
|
"galileo",
|
||||||
"braintrust",
|
"braintrust",
|
||||||
"arize",
|
"arize",
|
||||||
|
"gcs_bucket",
|
||||||
]
|
]
|
||||||
_known_custom_logger_compatible_callbacks: List = list(
|
_known_custom_logger_compatible_callbacks: List = list(
|
||||||
get_args(_custom_logger_compatible_callbacks_literal)
|
get_args(_custom_logger_compatible_callbacks_literal)
|
||||||
|
|
|
@ -61,6 +61,7 @@ from ..integrations.custom_logger import CustomLogger
|
||||||
from ..integrations.datadog import DataDogLogger
|
from ..integrations.datadog import DataDogLogger
|
||||||
from ..integrations.dynamodb import DyanmoDBLogger
|
from ..integrations.dynamodb import DyanmoDBLogger
|
||||||
from ..integrations.galileo import GalileoObserve
|
from ..integrations.galileo import GalileoObserve
|
||||||
|
from ..integrations.gcs_bucket import GCSBucketLogger
|
||||||
from ..integrations.greenscale import GreenscaleLogger
|
from ..integrations.greenscale import GreenscaleLogger
|
||||||
from ..integrations.helicone import HeliconeLogger
|
from ..integrations.helicone import HeliconeLogger
|
||||||
from ..integrations.lago import LagoLogger
|
from ..integrations.lago import LagoLogger
|
||||||
|
@ -2014,6 +2015,14 @@ def _init_custom_logger_compatible_class(
|
||||||
_langsmith_logger = LangsmithLogger()
|
_langsmith_logger = LangsmithLogger()
|
||||||
_in_memory_loggers.append(_langsmith_logger)
|
_in_memory_loggers.append(_langsmith_logger)
|
||||||
return _langsmith_logger # type: ignore
|
return _langsmith_logger # type: ignore
|
||||||
|
elif logging_integration == "gcs_bucket":
|
||||||
|
for callback in _in_memory_loggers:
|
||||||
|
if isinstance(callback, GCSBucketLogger):
|
||||||
|
return callback # type: ignore
|
||||||
|
|
||||||
|
_gcs_bucket_logger = GCSBucketLogger()
|
||||||
|
_in_memory_loggers.append(_gcs_bucket_logger)
|
||||||
|
return _gcs_bucket_logger # type: ignore
|
||||||
elif logging_integration == "arize":
|
elif logging_integration == "arize":
|
||||||
if "ARIZE_SPACE_KEY" not in os.environ:
|
if "ARIZE_SPACE_KEY" not in os.environ:
|
||||||
raise ValueError("ARIZE_SPACE_KEY not found in environment variables")
|
raise ValueError("ARIZE_SPACE_KEY not found in environment variables")
|
||||||
|
@ -2128,6 +2137,10 @@ def get_custom_logger_compatible_class(
|
||||||
for callback in _in_memory_loggers:
|
for callback in _in_memory_loggers:
|
||||||
if isinstance(callback, LangsmithLogger):
|
if isinstance(callback, LangsmithLogger):
|
||||||
return callback
|
return callback
|
||||||
|
elif logging_integration == "gcs_bucket":
|
||||||
|
for callback in _in_memory_loggers:
|
||||||
|
if isinstance(callback, GCSBucketLogger):
|
||||||
|
return callback
|
||||||
elif logging_integration == "otel":
|
elif logging_integration == "otel":
|
||||||
from litellm.integrations.opentelemetry import OpenTelemetry
|
from litellm.integrations.opentelemetry import OpenTelemetry
|
||||||
|
|
||||||
|
|
|
@ -55,4 +55,4 @@ general_settings:
|
||||||
max_response_size_mb: 10
|
max_response_size_mb: 10
|
||||||
|
|
||||||
litellm_settings:
|
litellm_settings:
|
||||||
success_callback: ["langfuse"]
|
callbacks: ["gcs_bucket"] # 👈 KEY CHANGE
|
Loading…
Add table
Add a link
Reference in a new issue