mojuss commited on
Commit
718ae1b
1 Parent(s): f7d5b3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -16
app.py CHANGED
@@ -57,23 +57,23 @@ def construct_index(directory_path):
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-1106", 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
 
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