apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-28 10:26:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

upgrade llama-index to v0.10

Browse source
This commit is contained in:
seehi 2024-02-23 11:06:53 +08:00 • committed by betterwang
parent 04527cf0eb
commit e14aedcea7
29 changed files with 725 additions and 370 deletions
Download patch file Download diff file Expand all files Collapse all files

5
metagpt/utils/embedding.py
View file

@ -5,12 +5,15 @@
@Author : alexanderwu
@File : embedding.py
"""
from llama_index.embeddings import OpenAIEmbedding
from llama_index.embeddings.openai import OpenAIEmbedding
from metagpt.config2 import config
def get_embedding() -> OpenAIEmbedding:
llm = config.get_openai_llm()
if llm is None:
raise ValueError("To use OpenAIEmbedding, please ensure that config.llm.api_type is correctly set to 'openai'.")
embedding = OpenAIEmbedding(api_key=llm.api_key, api_base=llm.base_url)
return embedding