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

update requirements.txt

Browse source
This commit is contained in:
seehi 2024-02-26 14:19:00 +08:00
parent 4264f8c1bd
commit 0a3c120806
2 changed files with 3 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

11
metagpt/rag/factories/llm.py
View file

@ -3,7 +3,6 @@
The LLM of LlamaIndex and the LLM of MG are not the same.
"""
from llama_index.core.llms import LLM
from llama_index.llms.anthropic import Anthropic
from llama_index.llms.azure_openai import AzureOpenAI
from llama_index.llms.gemini import Gemini
from llama_index.llms.ollama import Ollama
@ -21,7 +20,6 @@ class RAGLLMFactory(GenericFactory):
creators = {
LLMType.OPENAI: self._create_openai,
LLMType.AZURE: self._create_azure,
LLMType.ANTHROPIC: self._create_anthropic,
LLMType.GEMINI: self._create_gemini,
LLMType.OLLAMA: self._create_ollama,
}
@ -51,15 +49,6 @@ class RAGLLMFactory(GenericFactory):
temperature=config.llm.temperature,
)
def _create_anthropic(self):
return Anthropic(
base_url=config.llm.base_url,
api_key=config.llm.api_key,
model=config.llm.model,
max_tokens=config.llm.max_token,
temperature=config.llm.temperature,
)
def _create_gemini(self):
return Gemini(
api_base=config.llm.base_url,