Spaces:
Sleeping
Sleeping
arman77mxx
commited on
Commit
•
2315660
1
Parent(s):
d57edca
update
Browse files
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
|
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 =
|
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):
|