j6jddlmZmZmZddlmZddlmZedeZ d deed eegeffd Z y) )CallableOptionalTypeVar)litellm_completion_decorator)completion_chunks_aggregatorF)boundN project_namereturncptj}|jdd|tjS)aDecorator for tracking LiteLLM function calls with Opik. Can be used within other Opik-tracked functions to create proper span hierarchy. Supported (streaming and non-streaming modes): * `litellm.completion` * `litellm.acompletion` Example: ```python import litellm from opik.integrations.litellm import track_completion tracked_completion = track_completion(project_name="my-project")(litellm.completion) response = tracked_completion(model="gpt-3.5-turbo", messages=[...]) ``` Args: project_name: The name of the project to log data. Returns: Decorator function that wraps the completion function with Opik tracking. llmN)typenamer generations_aggregator)rLiteLLMCompletionTrackDecoratortrackr aggregate)r decorator_factorys /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/integrations/litellm/opik_tracker.pytrack_completionr s=65TTV  " "  !;EE # )N) typingrrrrrr strrrrrsF..** Cx #'"3-" qc1f"r