akman12914 commited on
Commit
eb4114a
Β·
1 Parent(s): e70c205

JSONLoader revise

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -55,13 +55,13 @@ def get_csv_file(docs):
55
  def get_json_file(docs):
56
  temp_dir = tempfile.TemporaryDirectory()
57
  temp_filepath = os.path.join(temp_dir.name, docs.name)
58
- with open(temp_filepath, 'wb') as f:
59
  f.write(docs.getvalue())
60
  json_loader = JSONLoader(file_path=temp_filepath,
61
  jq_schema='.messages[].content',
62
  text_content=False)
63
- json_doc = json_loader.load()
64
- return json_doc
65
 
66
 
67
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.
 
55
  def get_json_file(docs):
56
  temp_dir = tempfile.TemporaryDirectory()
57
  temp_filepath = os.path.join(temp_dir.name, docs.name)
58
+ with open(temp_filepath, "wb") as f:
59
  f.write(docs.getvalue())
60
  json_loader = JSONLoader(file_path=temp_filepath,
61
  jq_schema='.messages[].content',
62
  text_content=False)
63
+ data = json_loader.load()
64
+ return data
65
 
66
 
67
  # λ¬Έμ„œλ“€μ„ μ²˜λ¦¬ν•˜μ—¬ ν…μŠ€νŠΈ 청크둜 λ‚˜λˆ„λŠ” ν•¨μˆ˜μž…λ‹ˆλ‹€.