(self, message: str, stream=True, **model_kwargs)
| 62 | self.model_kwargs['model'] = model |
| 63 | |
| 64 | def _ask(self, message: str, stream=True, **model_kwargs): |
| 65 | if self.pending_stream_reply: |
| 66 | self._messages.append({"role": "assistant", "content": self.pending_stream_reply.result()}) |
| 67 | self.pending_stream_reply = None |
| 68 | |
| 69 | self._messages.append({"role": "user", "content": message}) |
| 70 | resp = self._client.chat.completions.create( |
| 71 | **self.model_kwargs, |
| 72 | **model_kwargs, |
| 73 | messages=self._messages, |
| 74 | stream=stream, |
| 75 | ) |
| 76 | return resp |
| 77 | |
| 78 | def ask(self, message: str, **model_kwargs) -> str: |
| 79 | """ |
no test coverage detected