Revert "Also remove Partner API section and examples"

This reverts commit 59bb6bd.

Author: OpheliaLjh

MIGRATION.md

@@ -7,7 +7,9 @@ This guide helps existing `oci-openai` users migrate to `oci-genai-auth`.
 - Replace dependency `oci-openai` with `oci-genai-auth`.
 - Continue using the `openai` Python SDK client.
 - Use `OciSessionAuth` to sign requests with OCI IAM.
-- Use endpoint `https://inference.generativeai.<region>.oci.oraclecloud.com/openai/v1` with a `project` OCID.
+- Choose endpoint/config based on API mode.:
+  - OCI Enterprise AI Agents: `https://inference.generativeai.<region>.oci.oraclecloud.com/openai/v1` + `project`
+  - Partner : `https://inference.generativeai.<region>.oci.oraclecloud.com/20231130/actions/v1` + `opc-compartment-id`
 
 ## 1) Dependency Changes
 
@@ -28,6 +30,8 @@ from oci_openai import OciSessionAuth
 
 ## 3) Client Initialization Changes
 
+### OCI Enterprise AI Agents
+
 Use the OpenAI-compatible endpoint and provide project OCID:
 
 ```python
@@ -43,7 +47,28 @@ client = OpenAI(
 )
 ```
 
+### Partner APIs
+
+Use `/v1` and include compartment header:
+
+```python
+import httpx
+from openai import OpenAI
+from oci_genai_auth import OciSessionAuth
+
+client = OpenAI(
+    base_url="https://inference.generativeai.<region>.oci.oraclecloud.com/v1",
+    api_key="not-used",
+    default_headers={"opc-compartment-id": "<ocid1.compartment...>"},
+    http_client=httpx.Client(auth=OciSessionAuth(profile_name="DEFAULT")),
+)
+```
+
 ## 4) Endpoint and required parameters
 
-- `base_url`: `https://inference.generativeai.<region>.oci.oraclecloud.com/openai/v1`
-- required: `project=<PROJECT_OCID>`
+- OCI Enterprise AI Agents:
+  - `base_url`: `https://inference.generativeai.<region>.oci.oraclecloud.com/openai/v1`
+  - required: `project=<PROJECT_OCID>`
+- Partner:
+  - `base_url`: `https://inference.generativeai.<region>.oci.oraclecloud.com/v1`
+  - required header: `opc-compartment-id=<COMPARTMENT_OCID>`

README.md

@@ -11,7 +11,8 @@ The **OCI GenAI Auth** Python library provides OCI request-signing helpers for t
 - [Using OCI IAM Auth](#using-oci-iam-auth)
 - [Using API Key Auth](#using-api-key-auth)
 - [Using OCI Enterprise AI Agents APIs](#using-oci-enterprise-ai-agents-apis)
-- [Examples](#examples)
+- [Using Partner APIs (passthrough)](#using-partner-apis-passthrough)
+- [Running the Examples](#running-the-examples)
 - [Contributing](#contributing)
 - [Security](#security)
 - [License](#license)
@@ -85,8 +86,33 @@ client = OpenAI(
 )
 ```
 
+## Using Partner APIs (passthrough)
+
+OCI also offers Partner API which passes through your calls to partners such as OpenAI. We will support more partners in the future.
+
+You can leverage Partner API when you want to use OpenAI's API and GPT models, but with OCI auth and billing.
+
+Note: Currently Partner API is only available to Oracle internal teams. Only features that meet partner's Zero Data Retention are available through Partner API.
+
+If you want multi-provider model access and features unavailable under partner's Zero Data Retention (such as File Search), use the OCI Enterprise AI Agents APIs above.
+
+```python
+import httpx
+from openai import OpenAI
+from oci_genai_auth import OciSessionAuth
+
+client = OpenAI(
+    base_url="https://inference.generativeai.us-chicago-1.oci.oraclecloud.com/20231130/actions/v1",
+    api_key="not-used",
+    default_headers={
+        "opc-compartment-id": "ocid1.compartment.oc1..aaaaaaaaexample",
+    },
+    http_client=httpx.Client(auth=OciSessionAuth(profile_name="DEFAULT")),
+)
+```
+
 ## Examples
-Demo code and instructions on how to run them can be found in the ```examples``` folder.
+Demo code and instructions on how to run them, for both OCI Enterprise AI Agents and partner usecases can be found in ```examples``` folder.
 
 ## Contributing
 

examples/partner/README.md

@@ -0,0 +1,29 @@
+# Partner Examples
+
+This folder contains partner API examples using the OpenAI Python SDK.
+
+## Prerequisites
+
+1. Install dependencies:
+
+   ```bash
+   pip install -e '.[dev]'
+   ```
+
+2. Configure shared values in `examples/partner/common.py`:
+   - `PROFILE_NAME`
+   - `COMPARTMENT_ID`
+   - `REGION`
+
+## How to run
+
+From repository root:
+
+```bash
+python -m examples.partner.openai.quickstart_openai_chat_completions
+```
+
+## Notes
+
+- Partner endpoints use pass-through mode and require the `opc-compartment-id` header.
+- These examples use IAM signing through `oci-genai-auth`.

examples/partner/__init__.py

@@ -0,0 +1,2 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/

examples/partner/common.py

@@ -0,0 +1,39 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+"""Partner example clients and configuration."""
+
+from __future__ import annotations
+
+import httpx
+from openai import AsyncOpenAI, OpenAI
+
+from oci_genai_auth import OciSessionAuth
+
+PROFILE_NAME = "DEFAULT"
+COMPARTMENT_ID = "<<ENTER_COMPARTMENT_ID>>"
+REGION = "us-chicago-1"
+
+PARTNER_OPENAI_BASE_URL = f"https://inference.generativeai.{REGION}.oci.oraclecloud.com/20231130/actions/v1"
+
+
+def build_openai_client() -> OpenAI:
+    client_kwargs = {
+        "api_key": "not-used",
+        "base_url": PARTNER_OPENAI_BASE_URL,
+        "http_client": httpx.Client(auth=OciSessionAuth(profile_name=PROFILE_NAME)),
+    }
+    if COMPARTMENT_ID:
+        client_kwargs["default_headers"] = {"opc-compartment-id": COMPARTMENT_ID}
+    return OpenAI(**client_kwargs)
+
+
+def build_openai_async_client() -> AsyncOpenAI:
+    client_kwargs = {
+        "api_key": "not-used",
+        "base_url": PARTNER_OPENAI_BASE_URL,
+        "http_client": httpx.AsyncClient(auth=OciSessionAuth(profile_name=PROFILE_NAME)),
+    }
+    if COMPARTMENT_ID:
+        client_kwargs["default_headers"] = {"opc-compartment-id": COMPARTMENT_ID}
+    return AsyncOpenAI(**client_kwargs)

examples/partner/openai/__init__.py

@@ -0,0 +1,2 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/

examples/partner/openai/basic_chat_completion.py

@@ -0,0 +1,29 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+"""Demonstrates a basic chat completion request for the Partner endpoint."""
+
+from rich import print
+
+from examples.partner import common
+
+MODEL = "openai.gpt-5.2"
+
+
+def main():
+    openai_client = common.build_openai_client()
+
+    completion = openai_client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {"role": "system", "content": "You are a concise assistant."},
+            {"role": "user", "content": "List three creative uses for a paperclip."},
+        ],
+        max_tokens=128,
+    )
+
+    print(completion.model_dump_json(indent=2))
+
+
+if __name__ == "__main__":
+    main()

examples/partner/openai/basic_chat_completion_api_key.py

@@ -0,0 +1,37 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+"""Demonstrates the Basic chat completion api key example."""
+
+import os
+
+from openai import OpenAI
+
+MODEL = "openai.gpt-5.2"
+
+
+def main() -> None:
+    client = OpenAI(
+        api_key=os.getenv("OPENAI_API_KEY"),
+        base_url="https://inference.generativeai.us-chicago-1.oci.oraclecloud.com/openai/v1",
+    )
+
+    response = client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {
+                "role": "system",
+                "content": "You are a concise assistant who answers in one paragraph.",
+            },
+            {
+                "role": "user",
+                "content": "Explain why the sky is blue as if you were a physics teacher.",
+            },
+        ],
+    )
+
+    print(response.choices[0].message.content)
+
+
+if __name__ == "__main__":
+    main()

examples/partner/openai/quickstart_openai_chat_completions.py

@@ -0,0 +1,46 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+"""Demonstrates a simple openai chat completions example."""
+
+import logging
+
+from examples.partner import common
+
+logging.basicConfig(level=logging.DEBUG)
+
+MODEL="openai.gpt-5.2"
+
+def main():
+    client = common.build_openai_client()
+
+    completion = client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {
+                "role": "user",
+                "content": "How do I output all files in a directory using Python?",
+            },
+        ],
+    )
+    print(completion.model_dump_json())
+
+    # Process the stream
+    print("=" * 80)
+    print("Process in streaming mode")
+    streaming = client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {
+                "role": "user",
+                "content": "How do I output all files in a directory using Python?",
+            },
+        ],
+        stream=True,
+    )
+    for chunk in streaming:
+        print(chunk)
+
+
+if __name__ == "__main__":
+    main()

examples/partner/openai/streaming_chat_completion.py

@@ -0,0 +1,38 @@
+# Copyright (c) 2026 Oracle and/or its affiliates.
+# Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
+
+"""Demonstrates streaming chat completion responses for the Partner (pass-through) endpoint."""
+
+from examples.partner import common
+
+MODEL = "openai.gpt-5.2"
+
+
+def main():
+    openai_client = common.build_openai_client()
+
+    stream = openai_client.chat.completions.create(
+        model=MODEL,
+        messages=[
+            {
+                "role": "system",
+                "content": "You are a concise assistant who answers in one paragraph.",
+            },
+            {
+                "role": "user",
+                "content": "Explain why the sky is blue as if you were a physics teacher.",
+            },
+        ],
+        stream=True,
+    )
+
+    for chunk in stream:
+        for choice in chunk.choices:
+            delta = choice.delta
+            if delta.content:
+                print(delta.content, end="", flush=True)
+    print()
+
+
+if __name__ == "__main__":
+    main()