Explicit model (langchain-ai#223)

arianpasquali · Dec 7, 2023 · e3329c2 · e3329c2
1 parent e6231f5
commit e3329c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/ingest.py b/ingest.py
@@ -52,6 +52,7 @@ def load_langchain_docs():
         meta_function=metadata_extractor,
     ).load()
 
+
 def load_langsmith_docs():
     return RecursiveUrlLoader(
         url="https://docs.smith.langchain.com/",
@@ -97,7 +98,7 @@ def load_api_docs():
 
 def get_embeddings_model() -> Embeddings:
     if os.environ.get("VOYAGE_AI_URL") and os.environ.get("VOYAGE_AI_MODEL"):
-        return VoyageEmbeddings()
+        return VoyageEmbeddings(model=os.environ["VOYAGE_AI_MODEL"])
     return OpenAIEmbeddings(chunk_size=200)
 
 

diff --git a/main.py b/main.py
@@ -213,7 +213,9 @@ def create_chain(
 )
 
 
-add_routes(app, answer_chain, path="/chat", input_type=ChatRequest, config_keys=["metadata"])
+add_routes(
+    app, answer_chain, path="/chat", input_type=ChatRequest, config_keys=["metadata"]
+)
 
 
 class SendFeedbackBody(BaseModel):