apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-21 14:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: replaced with OPTIONS

Browse source
This commit is contained in:
莘权 马 2023-08-28 17:45:45 +08:00
parent 3a1ebf19b7
commit 143ffb0c2c
39 changed files with 144 additions and 252 deletions
Download patch file Download diff file Expand all files Collapse all files

6
metagpt/tools/openai_text_to_embedding.py
View file

@ -17,8 +17,9 @@ import requests
from pydantic import BaseModel
import sys
from metagpt.config import CONFIG
sys.path.append(str(Path(__file__).resolve().parent.parent.parent)) # fix-bug: No module named 'metagpt'
from metagpt.utils.common import initialize_environment
from metagpt.logs import logger
@ -83,12 +84,11 @@ async def oas3_openai_text_to_embedding(text, model="text-embedding-ada-002", op
if not text:
return ""
if not openai_api_key:
openai_api_key = os.environ.get("OPENAI_API_KEY")
openai_api_key = CONFIG.OPENAI_API_KEY
return await OpenAIText2Embedding(openai_api_key).text_2_embedding(text, model=model)
if __name__ == "__main__":
initialize_environment()
loop = asyncio.new_event_loop()
v = loop.create_task(oas3_openai_text_to_embedding("Panda emoji"))
loop.run_until_complete(v)