commit_8

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='.', text_content=False)
     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_doc
+    # 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
 
 # 문서들을 처리하여 텍스트 청크로 나누는 함수입니다.
 def get_text_chunks(documents):