arman77mxx commited on
Commit
2315660
1 Parent(s): d57edca
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import streamlit as st
2
  import os
3
  from llama_index import SimpleDirectoryReader, Document, VectorStoreIndex
4
  from llama_index.node_parser import SimpleNodeParser
5
- from llama_index.query_engine import VectorIndexRetriever
6
  from llama_index.embeddings.openai import OpenAIEmbedding
7
  from llama_index.llms import OpenAI
8
  from llama_index.text_splitter import TokenTextSplitter
@@ -68,7 +68,7 @@ if OPENAI_API_KEY:
68
  llm = OpenAI(api_key=OPENAI_API_KEY, model_name="gpt-3.5-turbo-0125", temperature=0.8)
69
 
70
  # Configuración del retriever
71
- retriever = VectorIndexRetriever(index=vectorstore)
72
 
73
  # Definición del pipeline de procesamiento
74
  def rag_pipeline(question):
 
2
  import os
3
  from llama_index import SimpleDirectoryReader, Document, VectorStoreIndex
4
  from llama_index.node_parser import SimpleNodeParser
5
+ from llama_index.query_engine import SimpleRetriever
6
  from llama_index.embeddings.openai import OpenAIEmbedding
7
  from llama_index.llms import OpenAI
8
  from llama_index.text_splitter import TokenTextSplitter
 
68
  llm = OpenAI(api_key=OPENAI_API_KEY, model_name="gpt-3.5-turbo-0125", temperature=0.8)
69
 
70
  # Configuración del retriever
71
+ retriever = SimpleRetriever(index=vectorstore)
72
 
73
  # Definición del pipeline de procesamiento
74
  def rag_pipeline(question):