commit_6

app.py

@@ -11,13 +11,13 @@ from langchain.document_loaders import PyPDFLoader, TextLoader, JSONLoader, CSVL
 import tempfile # 임시 파일을 생성하기 위한 라이브러리입니다.
 import os
 from huggingface_hub import hf_hub_download # Hugging Face Hub에서 모델을 다운로드하기 위한 함수입니다.
-# from langchain.agents import (
-#     create_json_agent,
-#     AgentExecutor
-# )
-# from langchain.agents.agent_toolkits import JsonToolkit
-# from langchain.llms.openai import OpenAI
-# from langchain.tools.json.tool import JsonSpec
+from langchain.agents import (
+    create_json_agent,
+    AgentExecutor
+)
+from langchain.agents.agent_toolkits import JsonToolkit
+from langchain.llms.openai import OpenAI
+from langchain.tools.json.tool import JsonSpec
 
 # PDF 문서로부터 텍스트를 추출하는 함수입니다.
 def get_pdf_text(pdf_docs):
@@ -56,15 +56,15 @@ def get_json_file(docs):
         f.write(docs.getvalue())
     json_loader = JSONLoader(temp_filepath, jq_schema='.')
     json_data = json_loader.load()
-    # json_spec = JsonSpec(dict_=json_data, max_value_length=4000)
-    # json_toolkit = JsonToolkit(spec=json_spec)
-    # json_agent_executor = create_json_agent(
-    #     llm=OpenAI(temperature=0),
-    #     toolkit=json_toolkit,
-    #     verbose=True
-    # )
-    # json_doc = json_agent_executor.execute(text=temp_filepath)
-    return json_data
+    json_spec = JsonSpec(dict_=json_data, max_value_length=4000)
+    json_toolkit = JsonToolkit(spec=json_spec)
+    json_agent_executor = create_json_agent(
+        llm=OpenAI(temperature=0),
+        toolkit=json_toolkit,
+        verbose=True
+    )
+    json_doc = json_agent_executor.execute(text=temp_filepath)
+    return json_doc
 
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
 def get_text_chunks(documents):