Fixing missing OrgId assignment issue in the ChatCompletionStream (#268)

* Update client.go

* Add test

---------

Co-authored-by: panjiajia <panjiajia@haoshuo.com>
This commit is contained in:
biubiu7
2023-04-20 21:08:29 +08:00
committed by GitHub
parent ecdea45b67
commit c2b58e77ed
2 changed files with 17 additions and 0 deletions

View File

@@ -134,6 +134,9 @@ func (c *Client) newStreamRequest(
// OpenAI or Azure AD authentication
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.config.authToken))
}
if c.config.OrgID != "" {
req.Header.Set("OpenAI-Organization", c.config.OrgID)
}
return req, nil
}