alihan/genai-stack-neo4j
1
0
Fork 0
mirror of https://github.com/docker/genai-stack.git synced 2024-08-30 16:49:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #66 from docker/tomasonjo-patch-1

Browse Source 
Fix chains bug
This commit is contained in:
Tomaz Bratanic
2023-10-23 07:33:44 +02:00
committed by GitHub
parent 5883958799 8c4e67a6fc
commit 6022155d9d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
chains.py
View File

@@ -28,7 +28,7 @@ def load_embedding_model(embedding_model_name: str, logger=BaseLogger(), config=
embeddings = OpenAIEmbeddings()
dimension = 1536
logger.info("Embedding: Using OpenAI")
if embedding_model_name == "aws":
elif embedding_model_name == "aws":
embeddings = BedrockEmbeddings()
dimension = 1536
logger.info("Embedding: Using AWS")