Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import openai
|
|
6 |
import pinecone
|
7 |
from llama_index import VectorStoreIndex, StorageContext
|
8 |
from llama_index.chat_engine.types import ChatMode
|
9 |
-
from llama_index.llms import ChatMessage, MessageRole
|
10 |
from llama_index.vector_stores import PineconeVectorStore
|
11 |
|
12 |
from environments import OPENAI_API_KEY, PINECONE_API_KEY, PINECONE_INDEX, PASSWORD
|
@@ -20,9 +20,12 @@ pinecone.init(
|
|
20 |
)
|
21 |
pinecone_index = pinecone.Index(PINECONE_INDEX)
|
22 |
|
|
|
|
|
|
|
23 |
vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
|
24 |
storage_context = StorageContext.from_defaults(vector_store=vector_store)
|
25 |
-
index = VectorStoreIndex.from_documents([], storage_context=storage_context)
|
26 |
chat_engine = index.as_chat_engine(chat_mode=ChatMode.CONTEXT)
|
27 |
DENIED_ANSWER_PROMPT = '對不起,我是設計用於回答關於屯門地區康健中心的服務內容'
|
28 |
|
|
|
6 |
import pinecone
|
7 |
from llama_index import VectorStoreIndex, StorageContext
|
8 |
from llama_index.chat_engine.types import ChatMode
|
9 |
+
from llama_index.llms import ChatMessage, MessageRole, OpenAI
|
10 |
from llama_index.vector_stores import PineconeVectorStore
|
11 |
|
12 |
from environments import OPENAI_API_KEY, PINECONE_API_KEY, PINECONE_INDEX, PASSWORD
|
|
|
20 |
)
|
21 |
pinecone_index = pinecone.Index(PINECONE_INDEX)
|
22 |
|
23 |
+
llm = OpenAI(temperature=0.1, model="gpt-3.5-turbo-1106")
|
24 |
+
service_context = ServiceContext.from_defaults(llm=llm)
|
25 |
+
|
26 |
vector_store = PineconeVectorStore(pinecone_index=pinecone_index)
|
27 |
storage_context = StorageContext.from_defaults(vector_store=vector_store)
|
28 |
+
index = VectorStoreIndex.from_documents([], storage_context=storage_context, service_context=service_context)
|
29 |
chat_engine = index.as_chat_engine(chat_mode=ChatMode.CONTEXT)
|
30 |
DENIED_ANSWER_PROMPT = '對不起,我是設計用於回答關於屯門地區康健中心的服務內容'
|
31 |
|