Update app.py
Browse files
app.py
CHANGED
@@ -57,23 +57,23 @@ def construct_index(directory_path):
|
|
57 |
|
58 |
return index
|
59 |
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
|
68 |
-
def chatbotGPT(input):
|
69 |
-
if input:
|
70 |
-
messages.append({"role": "user", "content": input})
|
71 |
-
chat = openai.ChatCompletion.create(
|
72 |
-
model="gpt-3.5-turbo-
|
73 |
-
)
|
74 |
-
reply = chat.choices[0].message.content
|
75 |
-
messages.append({"role": "assistant", "content": reply})
|
76 |
-
return reply
|
77 |
|
78 |
def clear():
|
79 |
return None, None
|
|
|
57 |
|
58 |
return index
|
59 |
|
60 |
+
def chatbotCustom(input):
|
61 |
+
storage_context = StorageContext.from_defaults(persist_dir="index.json")
|
62 |
+
index = load_index_from_storage(storage_context)
|
63 |
+
query_engine = index.as_query_engine()
|
64 |
+
response = query_engine.query(input)
|
65 |
+
response = index.query(input, similarity_top_k=5, response_mode="tree_summarize")
|
66 |
+
return response.response
|
67 |
|
68 |
+
#def chatbotGPT(input):
|
69 |
+
# if input:
|
70 |
+
# messages.append({"role": "user", "content": input})
|
71 |
+
# chat = openai.ChatCompletion.create(
|
72 |
+
# model="gpt-3.5-turbo-instruct", messages=messages
|
73 |
+
# )
|
74 |
+
# reply = chat.choices[0].message.content
|
75 |
+
# messages.append({"role": "assistant", "content": reply})
|
76 |
+
# return reply
|
77 |
|
78 |
def clear():
|
79 |
return None, None
|