Mustehson commited on
Commit
b67984f
·
1 Parent(s): 666a38f

LLM Response

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -147,8 +147,8 @@ def process_inputs(inputs) :
147
  def run_llm(messages):
148
  try:
149
  response = llm.invoke(messages)
150
- print(response.content)
151
- tests = json.loads(response.content)
152
  except Exception as e:
153
  return e
154
  return tests
 
147
  def run_llm(messages):
148
  try:
149
  response = llm.invoke(messages)
150
+ print(response.content.replace("```", "'''").replace("json", ""))
151
+ tests = json.loads(response.content.replace("```", "").replace("json", ""))
152
  except Exception as e:
153
  return e
154
  return tests