(ctx context.Context, timeout time.Duration, logger logrus.FieldLogger, )
| 99 | } |
| 100 | |
| 101 | func (m *NvidiaModule) initVectorizer(ctx context.Context, timeout time.Duration, |
| 102 | logger logrus.FieldLogger, |
| 103 | ) error { |
| 104 | apiKey := os.Getenv("NVIDIA_APIKEY") |
| 105 | client := clients.New(apiKey, timeout, logger) |
| 106 | |
| 107 | m.vectorizer = text2vecbase.New(client, |
| 108 | batch.NewBatchVectorizer(client, 50*time.Second, batchSettings, logger, m.Name()), |
| 109 | batch.ReturnBatchTokenizer(batchSettings.TokenMultiplier, m.Name(), ent.LowerCaseInput), |
| 110 | ) |
| 111 | m.metaProvider = client |
| 112 | |
| 113 | return nil |
| 114 | } |
| 115 | |
| 116 | func (m *NvidiaModule) initAdditionalPropertiesProvider() error { |
| 117 | m.additionalPropertiesProvider = additional.NewText2VecProvider() |
no test coverage detected