- return &vectorizer{
- apiKey: apiKey,
- httpClient: &http.Client{
- Timeout: timeout,
- },
- urlBuilder: newCohereUrlBuilder(),
- logger: logger,
- }
- }
-
- func (v *vectorizer) Vectorize(ctx context.Context, input []string,
- cfg moduletools.ClassConfig,
- ) (*modulecomponents.VectorizationResult, *modulecomponents.RateLimits, error) {
- config := v.getVectorizationConfig(cfg)
- res, err := v.vectorize(ctx, input, config.Model, config.Truncate, config.BaseURL, searchDocument)
- return res, nil, err
- }
-
- func (v *vectorizer) VectorizeQuery(ctx context.Context, input []string,
- cfg moduletools.ClassConfig,
- ) (*modulecomponents.VectorizationResult, error) {