add milvus vector store

这个提交包含在:
binary-husky
2024-09-08 15:19:03 +00:00
父节点 e4e00b713f
当前提交 8b91d2ac0a
共有 6 个文件被更改,包括 128 次插入8 次删除

查看文件

@@ -71,7 +71,13 @@ class OpenAiEmbeddingModel(EmbeddingModel):
embedding = res.data[0].embedding
return embedding
def embedding_dimension(self, llm_kwargs):
def embedding_dimension(self, llm_kwargs=None):
# load kwargs
if llm_kwargs is None:
llm_kwargs = self.llm_kwargs
if llm_kwargs is None:
raise RuntimeError("llm_kwargs is not provided!")
from .bridge_all_embed import embed_model_info
return embed_model_info[llm_kwargs['embed_model']]['embed_dimension']