76 lines
2.1 KiB
Python
76 lines
2.1 KiB
Python
|
import importlib
|
||
|
|
import importlib.util
|
||
|
|
import logging
|
||
|
|
import os
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
from opentelemetry import context as context_api
|
||
|
|
from opentelemetry.instrumentation.google_generativeai.config import Config
|
||
|
|
|
||
|
|
TRACELOOP_TRACE_CONTENT = "TRACELOOP_TRACE_CONTENT"
|
||
|
|
|
||
|
|
|
||
|
|
def dont_throw(func):
|
||
|
|
"""
|
||
|
|
A decorator that wraps the passed in function and logs exceptions instead of throwing them.
|
||
|
|
|
||
|
|
@param func: The function to wrap
|
||
|
|
@return: The wrapper function
|
||
|
|
"""
|
||
|
|
# Obtain a logger specific to the function's module
|
||
|
|
logger = logging.getLogger(func.__module__)
|
||
|
|
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
try:
|
||
|
|
return func(*args, **kwargs)
|
||
|
|
except Exception as e:
|
||
|
|
logger.debug(
|
||
|
|
"OpenLLMetry failed to trace in %s, error: %s",
|
||
|
|
func.__name__,
|
||
|
|
traceback.format_exc(),
|
||
|
|
)
|
||
|
|
if Config.exception_logger:
|
||
|
|
Config.exception_logger(e)
|
||
|
|
|
||
|
|
return wrapper
|
||
|
|
|
||
|
|
|
||
|
|
def should_send_prompts():
|
||
|
|
return (
|
||
|
|
os.getenv(TRACELOOP_TRACE_CONTENT) or "true"
|
||
|
|
).lower() == "true" or context_api.get_value("override_enable_content_tracing")
|
||
|
|
|
||
|
|
|
||
|
|
def should_emit_events() -> bool:
|
||
|
|
"""
|
||
|
|
Checks if the instrumentation isn't using the legacy attributes
|
||
|
|
and if the event logger is not None.
|
||
|
|
"""
|
||
|
|
|
||
|
|
return not Config.use_legacy_attributes
|
||
|
|
|
||
|
|
|
||
|
|
def part_to_dict(part):
|
||
|
|
response = {}
|
||
|
|
|
||
|
|
if part.text:
|
||
|
|
response["text"] = part.text
|
||
|
|
if part.inline_data:
|
||
|
|
response["inline_data"] = part.inline_data
|
||
|
|
if part.function_call:
|
||
|
|
response["function_call"] = part.function_call
|
||
|
|
if part.function_response:
|
||
|
|
response["function_response"] = part.function_response
|
||
|
|
if part.file_data:
|
||
|
|
response["file_data"] = part.file_data
|
||
|
|
if part.executable_code:
|
||
|
|
response["executable_code"] = part.executable_code
|
||
|
|
if part.code_execution_result:
|
||
|
|
response["code_execution_result"] = part.code_execution_result
|
||
|
|
|
||
|
|
return response
|
||
|
|
|
||
|
|
|
||
|
|
def is_package_installed(package_name: str) -> bool:
|
||
|
|
return importlib.util.find_spec(package_name) is not None
|