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

add google gemini

Browse source
This commit is contained in:
better629 2023-12-14 16:45:40 +08:00
parent 183d939ee8
commit bef8d64193
7 changed files with 192 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
metagpt/llm.py
View file

@ -14,6 +14,7 @@ from metagpt.provider.spark_api import SparkAPI
from metagpt.provider.open_llm_api import OpenLLMGPTAPI
from metagpt.provider.fireworks_api import FireWorksGPTAPI
from metagpt.provider.human_provider import HumanProvider
from metagpt.provider.google_gemini_api import GeminiGPTAPI
def LLM() -> "BaseGPTAPI":
@ -29,6 +30,8 @@ def LLM() -> "BaseGPTAPI":
llm = OpenLLMGPTAPI()
elif CONFIG.fireworks_api_key:
llm = FireWorksGPTAPI()
elif CONFIG.gemini_api_key:
llm = GeminiGPTAPI()
else:
raise RuntimeError("You should config a LLM configuration first")