fix

sashabaranov · Nov 7, 2023 · 2354341 · 2354341
1 parent 934aef9
commit 2354341
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 12 deletions.
diff --git a/assistant.go b/assistant.go
@@ -10,7 +10,6 @@ import (
 const (
 	assistantsSuffix      = "/assistants"
 	assistantsFilesSuffix = "/files"
-	openaiBetaHeader      = "OpenAI-Beta"
 	openaiAssistantsV1    = "assistants=v1"
 )
 
@@ -79,7 +78,7 @@ type AssistantFilesList struct {
 // CreateAssistant creates a new assistant.
 func (c *Client) CreateAssistant(ctx context.Context, request AssistantRequest) (response Assistant, err error) {
 	req, err := c.newRequest(ctx, http.MethodPost, c.fullURL(assistantsSuffix), withBody(request),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -95,7 +94,7 @@ func (c *Client) RetrieveAssistant(
 ) (response Assistant, err error) {
 	urlSuffix := fmt.Sprintf("%s/%s", assistantsSuffix, assistantID)
 	req, err := c.newRequest(ctx, http.MethodGet, c.fullURL(urlSuffix),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -112,7 +111,7 @@ func (c *Client) ModifyAssistant(
 ) (response Assistant, err error) {
 	urlSuffix := fmt.Sprintf("%s/%s", assistantsSuffix, assistantID)
 	req, err := c.newRequest(ctx, http.MethodPost, c.fullURL(urlSuffix), withBody(request),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -128,7 +127,7 @@ func (c *Client) DeleteAssistant(
 ) (response Assistant, err error) {
 	urlSuffix := fmt.Sprintf("%s/%s", assistantsSuffix, assistantID)
 	req, err := c.newRequest(ctx, http.MethodDelete, c.fullURL(urlSuffix),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -166,7 +165,7 @@ func (c *Client) ListAssistants(
 
 	urlSuffix := fmt.Sprintf("%s%s", assistantsSuffix, encodedValues)
 	req, err := c.newRequest(ctx, http.MethodGet, c.fullURL(urlSuffix),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -184,7 +183,7 @@ func (c *Client) CreateAssistantFile(
 	urlSuffix := fmt.Sprintf("%s/%s%s", assistantsSuffix, assistantID, assistantsFilesSuffix)
 	req, err := c.newRequest(ctx, http.MethodPost, c.fullURL(urlSuffix),
 		withBody(request),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -201,7 +200,7 @@ func (c *Client) RetrieveAssistantFile(
 ) (response AssistantFile, err error) {
 	urlSuffix := fmt.Sprintf("%s/%s%s/%s", assistantsSuffix, assistantID, assistantsFilesSuffix, fileID)
 	req, err := c.newRequest(ctx, http.MethodGet, c.fullURL(urlSuffix),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -218,7 +217,7 @@ func (c *Client) DeleteAssistantFile(
 ) (err error) {
 	urlSuffix := fmt.Sprintf("%s/%s%s/%s", assistantsSuffix, assistantID, assistantsFilesSuffix, fileID)
 	req, err := c.newRequest(ctx, http.MethodDelete, c.fullURL(urlSuffix),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}
@@ -257,7 +256,7 @@ func (c *Client) ListAssistantFiles(
 
 	urlSuffix := fmt.Sprintf("%s/%s%s%s", assistantsSuffix, assistantID, assistantsFilesSuffix, encodedValues)
 	req, err := c.newRequest(ctx, http.MethodGet, c.fullURL(urlSuffix),
-		withHeader(openaiBetaHeader, openaiAssistantsV1))
+		withBetaAssistantV1())
 	if err != nil {
 		return
 	}

diff --git a/client.go b/client.go
@@ -83,9 +83,9 @@ func withContentType(contentType string) requestOption {
 	}
 }
 
-func withHeader(key, value string) requestOption {
+func withBetaAssistantV1() requestOption {
 	return func(args *requestOptions) {
-		args.header.Set(key, value)
+		args.header.Set("OpenAI-Beta", "assistants=v1")
 	}
 }