fix: add OCI tracer to init

Author: gustavocidornelas

src/openlayer/lib/__init__.py

@@ -11,6 +11,7 @@
     "trace_async_openai",
     "trace_async",
     "trace_bedrock",
+    "trace_oci",
 ]
 
 # ---------------------------------- Tracing --------------------------------- #
@@ -95,11 +96,29 @@ 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(client):
+    """Trace OCI queries."""
+    # pylint: disable=import-outside-toplevel
+    import oci
+    from .integrations import oci_tracer
+
+    if not isinstance(client, oci.generative_ai_inference.GenerativeAiInferenceClient):
+        raise ValueError("Invalid client. Please provide an OCI Generative AI client.")
+    return oci_tracer.trace_oci_genai(client)