Update stack

README.md

@@ -4,7 +4,7 @@ emoji: 👁
 colorFrom: red
 colorTo: purple
 sdk: gradio
-sdk_version: 4.16.0
+sdk_version: 4.29.0
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -1,25 +1,26 @@
 import os
 import json
 import gradio as gr
-from llama_index import (
+import chromadb
+
+from llama_index.core import (
     VectorStoreIndex,
+    StorageContext,
+    ServiceContext,
     download_loader,
 )
-import chromadb
 
-from llama_index.llms import MistralAI
-from llama_index.embeddings import MistralAIEmbedding
-from llama_index.vector_stores import ChromaVectorStore
-from llama_index.storage.storage_context import StorageContext
-from llama_index import ServiceContext
+from llama_index.llms.mistralai import MistralAI
+from llama_index.embeddings.mistralai import MistralAIEmbedding
+from llama_index.vector_stores.chroma import ChromaVectorStore
 
-title = "Gaia Mistral Chat RAG PDF Demo"
+title = "Gaia Mistral 8x7b Chat RAG PDF Demo"
 description = "Example of an assistant with Gradio, RAG from PDF documents and Mistral AI via its API"
 placeholder = (
     "Vous pouvez me posez une question sur ce contexte, appuyer sur Entrée pour valider"
 )
 placeholder_url = "Extract text from this url"
-llm_model = "mistral-small"
+llm_model = "open-mixtral-8x7b"
 
 env_api_key = os.environ.get("MISTRAL_API_KEY")
 query_engine = None

requirements.txt

@@ -1,5 +1,10 @@
-pypdf==4.0.1
-mistralai==0.0.12
-llama-index==0.9.40
-gradio==4.16.0
-chromadb==0.4.22
+pypdf
+mistralai
+gradio
+chromadb
+llama-index
+llama-index-readers-web
+llama-index-readers-file
+llama-index-llms-mistralai
+llama-index-embeddings-mistralai
+llama-index-vector-stores-chroma