Update app.py

app.py

@@ -40,14 +40,10 @@ class StreamingAgentCallbackHandler(BaseCallbackHandler):
     def on_llm_error(self, error, **kwargs):
         self.q.put(f"[ERROR] {str(error)}")
 
-#from langchain_community.retrievers import WikipediaRetriever
-#from langchain.tools.retriever import create_retriever_tool
-#retriever = WikipediaRetriever(lang="ko",top_k_results=10)
-#retriever_tool = create_retriever_tool(
-    #retriever,
-   # name="wiki_search",
-   # description="위키백과에서 필요한 정보를 불러옵니다.결괴를 검증하여 사용하시오.",
-#)
+from langchain_community.retrievers import WikipediaRetriever
+from langchain.tools.retriever import create_retriever_tool
+retriever = WikipediaRetriever(lang="ko",top_k_results=10)
+wiki=Tool(retriever.get_relevantt_documents,name="WIKI SEARCH",description="위키백과에서 필요한 정보를 불러옵니다.결괴를 검증하여 사용하시오.")
 # ──────────────────────────────
 # ✅ GitHub Models LLM
 # ──────────────────────────────
@@ -90,8 +86,7 @@ llm = GitHubModelLLM(model="openai/gpt-4.1", token=token)
 tools = load_tools(
     ["ddg-search", "requests_all", "llm-math"],
     llm=llm,allow_dangerous_tools=True
-)+[YTS()]
-
+)+[YTS()]+[wiki]
 # ──────────────────────────────
 # ✅ Python 실행 도구 (LangChain 내장)
 # ──────────────────────────────
@@ -165,7 +160,9 @@ tools.extend([
     Tool(name="file_delete", func=file_delete, description="파일을 삭제합니다."),
     Tool(name="time_now", func=time_now, description="현재 시간을 반환합니다."),
 ])
-
+from langchain.memory import ConversationBufferMemory as MEM
+from langchain.agents.agent_toolkits import FileManagementToolkit as FMT
+tools.extend(FMT(root_dir=str(os.getcwd())).get_tools())
 # ──────────────────────────────
 # ✅ Agent 초기화
 # ──────────────────────────────
@@ -174,6 +171,7 @@ handler = StreamingAgentCallbackHandler(q_stream)
 agent = initialize_agent(
     tools,
     llm,
+    memory=MEM(),
     agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
     verbose=True,
     handle_parsing_errors=True,
@@ -194,7 +192,6 @@ def chat(message, history):
         if msg == "[END]":
             break
         yield msg + "\n"
-    return history, history
 
 with gr.Blocks(theme=gr.themes.Soft(), title="PIXAL Assistant (LangChain + GitHub LLM)") as demo:
     gr.Markdown("""