fix: add OCI tracer to init (#498)

* fix: add OCI tracer to init

* Update src/openlayer/lib/__init__.py

---------

Co-authored-by: Vinícius Mello <[email protected]>

Author: gustavocidornelas

src/openlayer/lib/__init__.py

@@ -11,6 +11,7 @@
     "trace_async_openai",
     "trace_async",
     "trace_bedrock",
+    "trace_oci",
 ]
 
 # ---------------------------------- Tracing --------------------------------- #
@@ -95,11 +96,40 @@ def trace_bedrock(client):
     try:
         import boto3
     except ImportError:
-        raise ImportError("boto3 is required for Bedrock tracing. Install with: pip install boto3")
+        raise ImportError(
+            "boto3 is required for Bedrock tracing. Install with: pip install boto3"
+        )
 
     from .integrations import bedrock_tracer
 
     # Check if it's a boto3 client for bedrock-runtime service
-    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.")
+    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)