cowcow02 commited on
Commit
2fde223
1 Parent(s): ba130f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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