update requirements.txt

This commit is contained in:
seehi 2024-02-26 14:19:00 +08:00 committed by betterwang
parent 65f85b2bdd
commit 6e405752b9
2 changed files with 3 additions and 15 deletions

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,

View file

@ -11,9 +11,8 @@ typer==0.9.0
# godot==0.1.1
# google_api_python_client==2.93.0 # Used by search_engine.py
lancedb==0.4.0
llama-index-core==0.10.11.post1
llama-index-core==0.10.12
llama-index-embeddings-openai==0.1.5
llama-index-llms-anthropic==0.1.3
llama-index-llms-azure-openai==0.1.4
llama-index-llms-gemini==0.1.4
llama-index-llms-ollama==0.1.2
@ -26,7 +25,7 @@ meilisearch==0.21.0
numpy==1.24.3
openai==1.6.1
openpyxl
beautifulsoup4==4.12.2
beautifulsoup4==4.12.3
pandas==2.0.3
pydantic==2.5.3
#pygame==2.1.3
@ -38,7 +37,7 @@ PyYAML==6.0.1
setuptools==65.6.3
tenacity==8.2.3
tiktoken==0.5.2
tqdm==4.65.0
tqdm==4.66.2
#unstructured[local-inference]
# selenium>4
# webdriver_manager<3.9