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

Update initialize_game to include user input

Browse files
Files changed (1) hide show
  1. chat.py +24 -10
chat.py CHANGED
@@ -133,8 +133,29 @@ def populate_prompt(game_id):
133
  return whole_prompt
134
 
135
 
136
- def initialize_game(user_id, game_id):
137
- whole_prompt = populate_prompt(game_id)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
138
  llm_prompt_op = call_gpt(whole_prompt)
139
  #print(llm_prompt_op.choices[0]["message"]["content"])
140
  fname="prompt_" + game_id + "_" + user_id + ".txt"
@@ -142,11 +163,4 @@ def initialize_game(user_id, game_id):
142
  return llm_prompt_op.choices[0]["message"]["content"]
143
 
144
  if __name__ == '__main__':
145
- user_id='user_1'
146
- game_id = '536e6bc89df5'
147
- output = initialize_game(user_id, game_id)
148
- print('\n\nGENESIS: %s' % output)
149
- while True:
150
- a = input('\n\n%s: ' % user_id)
151
- output = start_game(game_id, user_id, user_input=a)
152
- print('\n\nGENESIS: %s' % output)
 
133
  return whole_prompt
134
 
135
 
136
+ def populate_prompt(game_id, splits):
137
+ prompt_text = list()
138
+ idlist = []
139
+ for j in range(int(splits)):
140
+ idlist.append(game_id + "-" + str(j))
141
+
142
+ results=vector_db.fetch(ids=idlist)
143
+ for ids in idlist:
144
+ prompt_text.append(results['vectors'][ids]["metadata"]["text"])
145
+
146
+ whole_prompt = ' '.join(prompt_text).strip()
147
+ return whole_prompt
148
+
149
+
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"
 
163
  return llm_prompt_op.choices[0]["message"]["content"]
164
 
165
  if __name__ == '__main__':
166
+ print("main")