refactor: Refactor endpoint and model compatibility check (#180)

* Add model check for chat stream

* Sync model checks

* Fix typo

* Fix functino

* refactor: Refactor endpoint and model compatibility check

* apply review suggestions

* minor fix

* invert return boolean flag

* fix test
This commit is contained in:
Jo
2023-03-22 21:46:08 +08:00
committed by GitHub
parent 428839400a
commit 2ebb265e71
8 changed files with 94 additions and 13 deletions

View File

@@ -37,8 +37,14 @@ func (c *Client) CreateChatCompletionStream(
ctx context.Context,
request ChatCompletionRequest,
) (stream *ChatCompletionStream, err error) {
urlSuffix := "/chat/completions"
if !checkEndpointSupportsModel(urlSuffix, request.Model) {
err = ErrChatCompletionInvalidModel
return
}
request.Stream = true
req, err := c.newStreamRequest(ctx, "POST", "/chat/completions", request)
req, err := c.newStreamRequest(ctx, "POST", urlSuffix, request)
if err != nil {
return
}