groq

apocas · Feb 26, 2024 · 2890352 · 2890352
1 parent de2e3a0
commit 2890352
Show file tree

Hide file tree

Showing 3 changed files with 81 additions and 30 deletions.
diff --git a/app/tools.py b/app/tools.py
@@ -32,6 +32,9 @@ def getLLMClass(llm_classname):
  elif llm_classname == "OpenAI":
  from llama_index.llms.openai import OpenAI
  return OpenAI
+ elif llm_classname == "Groq":
+ from llama_index.llms.groq import Groq
+ return Groq
  else:
  raise Exception("Invalid LLM class name.")
 

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,16 +51,17 @@ opencv-python = "^4.9.0.80"
 ip-adapter = "^0.1.0"
 ilock = "^1.0.3"
 docx2txt = "^0.8"
-llama-index-core = "^0.10.3"
-llama-index-embeddings-langchain = "^0.1.1"
-llama-index-vector-stores-redis = "^0.1.1"
-llama-index-vector-stores-chroma = "^0.1.1"
+llama-index-core = "^0.10.12"
+llama-index-embeddings-langchain = "^0.1.2"
+llama-index-vector-stores-redis = "^0.1.2"
+llama-index-vector-stores-chroma = "^0.1.4"
 redisvl = {git = "https://github.com/RedisVentures/redisvl.git"}
 chromadb = "^0.4.22"
-llama-index-llms-openai = "^0.1.1"
+llama-index-llms-openai = "^0.1.6"
 einops = "^0.7.0"
-llama-index-llms-ollama = "^0.1.1"
+llama-index-llms-ollama = "^0.1.2"
 llama-index-postprocessor-colbert-rerank = "^0.1.1"
+llama-index-llms-groq = "^0.1.3"
 
 [tool.poetry.group.gpu]
 optional = true