Hatman commited on
Commit
fe710dc
·
verified ·
1 Parent(s): 4b351b2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -117,6 +117,9 @@ def inferencePrompt(item: Core):
117
  try:
118
  plain_response_data = getPrompt(item.itemString, prompt_model)
119
  magic_response_data = getPrompt(item.itemString, magic_prompt_model)
 
 
 
120
  plain_response = plain_response_data['choices'][0]['message']['content']
121
  returnJson = {"plain": plain_response, "magic": item.itemString + magic_response_data[0]["generated_text"]}
122
  print(f'Return Json {returnJson}')
 
117
  try:
118
  plain_response_data = getPrompt(item.itemString, prompt_model)
119
  magic_response_data = getPrompt(item.itemString, magic_prompt_model)
120
+ print(plain_response_data['choices'])
121
+ print(plain_response_data['choices'][0])
122
+ print(plain_response_data['choices'][0]['message'])
123
  plain_response = plain_response_data['choices'][0]['message']['content']
124
  returnJson = {"plain": plain_response, "magic": item.itemString + magic_response_data[0]["generated_text"]}
125
  print(f'Return Json {returnJson}')