diff --git a/pinecone/client.go b/pinecone/client.go index 924d742..883ae64 100644 --- a/pinecone/client.go +++ b/pinecone/client.go @@ -1325,7 +1325,7 @@ type EmbedRequest struct { // InputType string // Truncate string // } -type EmbedParameters *map[string]interface{} +type EmbedParameters map[string]interface{} // [EmbedResponse] represents holds the embeddings generated for a single input. // @@ -1407,17 +1407,9 @@ func (i *InferenceService) Embed(ctx context.Context, in *EmbedRequest) (*EmbedR } // convert embedding parameters to expected type - // if in.Parameters.InputType != "" || in.Parameters.Truncate != "" { - // req.Parameters = &struct { - // InputType string `json:"input_type,omitempty"` - // Truncate string `json:"truncate,omitempty"` - // }{ - // InputType: in.Parameters.InputType, - // Truncate: in.Parameters.Truncate, - // } - // } if &in.Parameters != nil { - req.Parameters = in.Parameters + params := map[string]interface{}(in.Parameters) + req.Parameters = ¶ms } res, err := i.client.Embed(ctx, req) diff --git a/pinecone/client_test.go b/pinecone/client_test.go index 88a9cfa..3335729 100644 --- a/pinecone/client_test.go +++ b/pinecone/client_test.go @@ -310,7 +310,7 @@ func (ts *IntegrationTests) TestGenerateEmbeddings() { "The quick brown fox jumps over the lazy dog", "Lorem ipsum", }, - Parameters: &map[string]interface{}{ + Parameters: map[string]interface{}{ "input_type": "query", "truncate": "END", }, @@ -328,7 +328,7 @@ func (ts *IntegrationTests) TestGenerateEmbeddingsInvalidInputs() { embeddingModel := "multilingual-e5-large" _, err := ts.client.Inference.Embed(ctx, &EmbedRequest{ Model: embeddingModel, - Parameters: &map[string]interface{}{ + Parameters: map[string]interface{}{ "input_type": "query", "truncate": "END", },