Add api client code

search.go

@@ -0,0 +1,42 @@
+package gogpt
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"net/http"
+)
+
+type SearchRequest struct {
+	Documents []string `json:"documents"`
+	Query     string   `json:"query"`
+}
+
+type SearchResult struct {
+	Document int     `json:"document"`
+	Score    float32 `json:"score"`
+}
+
+type SearchResponse struct {
+	SearchResults []SearchResult `json:"data"`
+}
+
+// Search — perform a semantic search api call over a list of documents.
+func (c *Client) Search(ctx context.Context, engineID string, request SearchRequest) (response SearchResponse, err error) {
+	var reqBytes []byte
+	reqBytes, err = json.Marshal(request)
+	if err != nil {
+		return
+	}
+
+	urlSuffix := fmt.Sprintf("/engines/%s/search", engineID)
+	req, err := http.NewRequest("POST", c.fullURL(urlSuffix), bytes.NewBuffer(reqBytes))
+	if err != nil {
+		return
+	}
+
+	req = req.WithContext(ctx)
+	err = c.sendRequest(req, &response)
+	return
+}