| 134 | |
| 135 | |
| 136 | class ChatWrapper(Chat): |
| 137 | def __init__( |
| 138 | self, |
| 139 | client: OriginalOpenAI, |
| 140 | openpipe_reporting_client: OpenPipe, |
| 141 | openpipe_completions_client: OriginalOpenAI, |
| 142 | ) -> None: |
| 143 | super().__init__(client) |
| 144 | self.completions = CompletionsWrapper( |
| 145 | client, openpipe_reporting_client, openpipe_completions_client |
| 146 | ) |
| 147 | |
| 148 | |
| 149 | class OpenAIWrapper(OriginalOpenAI): |