feat(tracing): add OCI GenAI tracing function

- Updated the `__init__.py` file to include the new tracing function in the module's exports, improving accessibility for users.

Author: viniciusdsmello

src/openlayer/lib/__init__.py

@@ -6,12 +6,13 @@
     "trace_anthropic",
     "trace_openai",
     "trace_openai_assistant_thread_run",
+    "trace_oci_genai",
     "trace_mistral",
     "trace_groq",
     "trace_async_openai",
     "trace_async",
     "trace_bedrock",
-]
+]   
 
 # ---------------------------------- Tracing --------------------------------- #
 from .tracing import tracer
@@ -103,3 +104,23 @@ def trace_bedrock(client):
     if not hasattr(client, "_service_model") or client._service_model.service_name != "bedrock-runtime":
         raise ValueError("Invalid client. Please provide a boto3 bedrock-runtime client.")
     return bedrock_tracer.trace_bedrock(client)
+
+def trace_oci_genai(client, estimate_tokens: bool = True):
+    """Trace OCI GenAI chat completions.
+    
+    Args:
+        client: OCI GenAI client.
+        estimate_tokens: Whether to estimate tokens when not available. Defaults to True.
+    """
+    # pylint: disable=import-outside-toplevel
+    try:
+        import oci
+    except ImportError:
+        raise ImportError("oci is required for OCI GenAI tracing. Install with: pip install oci")
+
+    from .integrations import oci_tracer
+
+    if not isinstance(client, oci.generative_ai_inference.GenerativeAiInferenceClient):
+        raise ValueError("Invalid client. Please provide an OCI GenAI client.")
+
+    return oci_tracer.trace_oci_genai(client, estimate_tokens=estimate_tokens)