laxsvips commited on
Commit
bb2e116
·
1 Parent(s): 4b0190f

Update chat.py

Browse files
Files changed (1) hide show
  1. chat.py +1 -20
chat.py CHANGED
@@ -116,23 +116,8 @@ def start_game(game_id, user_id, user_input):
116
  vector_db.upsert(payload)
117
 
118
  return(llm_output)
119
-
120
- def populate_prompt(game_id):
121
- prompt_text = list()
122
- idlist = []
123
- for j in range(21):
124
- idlist.append(game_id + "-" + str(j))
125
 
126
 
127
- results=vector_db.fetch(ids=idlist)
128
-
129
- for ids in idlist:
130
- prompt_text.append(results['vectors'][ids]["metadata"]["text"])
131
-
132
- whole_prompt = ' '.join(prompt_text).strip()
133
- return whole_prompt
134
-
135
-
136
  def populate_prompt(game_id, splits):
137
  prompt_text = list()
138
  idlist = []
@@ -150,12 +135,8 @@ def populate_prompt(game_id, splits):
150
  def initialize_game(game_id, user_id, user_input):
151
  game_details = get_game_details(game_id)
152
  whole_prompt = populate_prompt(game_id, game_details["splits"])
153
- if debug:
154
- print(whole_prompt[:1000])
155
  whole_prompt = whole_prompt.replace("<<USER_INPUT_MSG>>", user_input)
156
- if debug:
157
- print(whole_prompt[:1000])
158
-
159
  llm_prompt_op = call_gpt(whole_prompt)
160
  #print(llm_prompt_op.choices[0]["message"]["content"])
161
  fname="prompt_" + game_id + "_" + user_id + ".txt"
 
116
  vector_db.upsert(payload)
117
 
118
  return(llm_output)
 
 
 
 
 
 
119
 
120
 
 
 
 
 
 
 
 
 
 
121
  def populate_prompt(game_id, splits):
122
  prompt_text = list()
123
  idlist = []
 
135
  def initialize_game(game_id, user_id, user_input):
136
  game_details = get_game_details(game_id)
137
  whole_prompt = populate_prompt(game_id, game_details["splits"])
 
 
138
  whole_prompt = whole_prompt.replace("<<USER_INPUT_MSG>>", user_input)
139
+
 
 
140
  llm_prompt_op = call_gpt(whole_prompt)
141
  #print(llm_prompt_op.choices[0]["message"]["content"])
142
  fname="prompt_" + game_id + "_" + user_id + ".txt"