Fix missing embedding endpoint in patch_openai_client, unify test (tensorzero#3193)

* added missing endpoint

* set up OpenAI python tests to work with patched client

* removed all references to clients/openai-python

* added pyright ignore in test fixture

Author: virajmehta

.github/workflows/general.yml

@@ -305,18 +305,6 @@ jobs:
         run: |
           uv run stubtest tensorzero.tensorzero
 
-      - name: "Python: OpenAI Client: Install dependencies"
-        working-directory: clients/openai-python
-        run: |
-          uv venv
-          uv pip sync requirements.txt
-
-      - name: "Python: OpenAI Client: pyright"
-        working-directory: clients/openai-python
-        run: |
-          uv pip install pyright==1.1.394
-          uv run pyright
-
       - name: "Node.js: Run prettier"
         run: pnpm --filter=openai-node run format
 

.github/workflows/merge-queue.yml

@@ -238,18 +238,6 @@ jobs:
           fi
           exit 1
 
-      - name: "Python: OpenAI Client: Install dependencies"
-        working-directory: clients/openai-python
-        run: |
-          uv venv
-          uv pip sync requirements.txt
-
-      - name: "Python: OpenAI Client: pytest"
-        working-directory: clients/openai-python
-        timeout-minutes: 5
-        run: |
-          uv run pytest --verbose
-
       - name: "Node.js: OpenAI Client: test"
         working-directory: clients/openai-node
         run: |

.pre-commit-config.yaml

@@ -174,13 +174,6 @@ repos:
   - repo: https://github.com/RobertCraigie/pyright-python
     rev: v1.1.391
     hooks:
-      - id: pyright
-        name: "pyright: clients/openai-python"
-        files: ^clients/openai-python/
-        args: ["--project", "clients/openai-python/pyrightconfig.json"]
-        additional_dependencies:
-          - openai
-
       - id: pyright
         name: "pyright: recipes"
         files: ^recipes/

clients/README.md

@@ -1,5 +1,5 @@
 # TensorZero Gateway Clients
 
-The TensorZero Gateway can be used with the **TensorZero Python client** (see `./python`), with **OpenAI clients (see `./openai-python`, `./openai-node`, and `./openai-go` for examples)**, or via its **HTTP API in any programming language**.
+The TensorZero Gateway can be used with the **TensorZero Python client** (see `./python`), with **OpenAI clients (see `./python/tests`, `./openai-node`, and `./openai-go` for examples)**, or via its **HTTP API in any programming language**.
 
 Refer to [TensorZero Gateway Clients](https://www.tensorzero.com/docs/gateway/clients) in our documentation for usage details and more information.

clients/openai-python/pyproject.toml

@@ -26,6 +26,8 @@ uv run python
 
 Integration tests can be run with `./test.sh` (this requires the same setup as `cargo test-e2e` - see `CONTRIBUTING.md`)
 
+This also runs all tests for OpenAI compatibility in Python.
+
 ## Naming
 
 There are several different names in use in this client: