changed EmbeddingsGeneratorProvider env variables (#9)

theam · Jul 26, 2023 · 52176e1 · 52176e1
1 parent 9777553
commit 52176e1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/apps/semantic_search/embeddings/provider.py b/apps/semantic_search/embeddings/provider.py
@@ -10,14 +10,14 @@ class EmbeddingsGeneratorProvider:
     def get_client(name: str) -> EmbeddingsGenerator:
         if name == 'SPACES_TEXT':
             return SpacesGradioTextEmbeddings(
-                endpoint_url=get_env_or_fail("EMBEDDINGS_GENERATION_URL"),
-                endpoint_key=get_env_or_fail("EMBEDDINGS_GENERATION_KEY"),
+                endpoint_url=get_env_or_fail("SPACES_URL"),
+                endpoint_key=get_env_or_fail("SPACES_KEY"),
                 batch_size=int(get_env_or_fail("BATCH_SIZE")),
             )
         if name == 'SPACES_INSTRUCT':
             return SpacesGradioInstructTextEmbeddings(
-                endpoint_url=get_env_or_fail("EMBEDDINGS_GENERATION_URL"),
-                endpoint_key=get_env_or_fail("EMBEDDINGS_GENERATION_KEY"),
+                endpoint_url=get_env_or_fail("SPACES_URL"),
+                endpoint_key=get_env_or_fail("SPACES_KEY"),
                 batch_size=int(get_env_or_fail("BATCH_SIZE")),
             )
         if name == 'OPENAI':