Skip to content

Commit

Permalink
Merge pull request #468 from aws-samples/lvn
Browse files Browse the repository at this point in the history
fix: update embedding condition
  • Loading branch information
NingLu authored Dec 4, 2024
2 parents 0086d9d + 41ae3cd commit 4249f1c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion source/lambda/intention/intention.py
Original file line number Diff line number Diff line change
Expand Up @@ -418,7 +418,7 @@ def convert_qa_list(qa_list: list, bucket: str, prefix: str) -> List[Document]:

def __save_2_aos(modelId: str, index: str, qaListParam: list, bucket: str, prefix: str):
qaList = __deduplicate_by_key(qaListParam, "question")
if kb_enabled:
if kb_enabled and embedding_model_endpoint.startswith("bce-embedding"):
embedding_info = get_embedding_info(embedding_model_endpoint)
embedding_function = sm_utils.getCustomEmbeddings(
embedding_model_endpoint, region, embedding_info.get("ModelType")
Expand Down

0 comments on commit 4249f1c

Please sign in to comment.