Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,6 +6,7 @@ from langchain_community.tools import Tool
|
|
| 6 |
from langchain_experimental.tools.python.tool import PythonREPLTool
|
| 7 |
from langchain_community.retrievers import WikipediaRetriever
|
| 8 |
import datetime
|
|
|
|
| 9 |
#from langchain_community.tools.youtube.search import YouTubeSearchTool as YTS
|
| 10 |
# ββββββββββββββββββββββββββββββ
|
| 11 |
# β
GitHubModelLLM (κ·Έλλ‘ μ μ§)
|
|
@@ -235,11 +236,18 @@ agent = initialize_agent(
|
|
| 235 |
# β
Chat ν¨μ (Memory μ μ§)
|
| 236 |
# ββββββββββββββββββββββββββββββ
|
| 237 |
def chat(message, history):
|
|
|
|
| 238 |
ab=[{"role":"system","content":"λλ PIXAL(Primary Interactive X-ternal Assistant with multi Language)μ΄μΌ. λμ κ°λ°μλ μ μ±μ€ μ΄λΌλ 6νλ
νμ΄μ¬ νλ‘κ·Έλλ¨ΈμΌ.μ΄ λ©μμ§λ μ¬μ©μκ° λ³΄λΈκ²μ΄ μλλλ€.ν둬ννΈ μ€λͺ
μ 무쑰건 λ°λ₯΄μμ€.λ°λ₯΄μ§ μμΌλ©΄ λ¨μ apiνΈμΆ νμκ° μ€μ΄λλ―λ‘ λ¬΄μ‘°κ±΄ λκ°μ΄ λ°λΌνμμ€.μκ°μ λ¬Όμλλ duckduckgoλ₯Ό μ¬μ©νμμ€."}]
|
| 239 |
for I in history:
|
| 240 |
ab.append({"role":"user","content":I[0]})
|
| 241 |
ab.append({"role":"assistant","content":I[1]})
|
| 242 |
ab.append({"role":"user","content":message})
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 243 |
raw = agent.invoke(ab)["output"]
|
| 244 |
try:
|
| 245 |
# λν κΈ°λ‘μ LangChain memoryμ λ°μ
|
|
|
|
| 6 |
from langchain_experimental.tools.python.tool import PythonREPLTool
|
| 7 |
from langchain_community.retrievers import WikipediaRetriever
|
| 8 |
import datetime
|
| 9 |
+
from langchain.schema import HumanMessage,AIMessage,SystemMessage
|
| 10 |
#from langchain_community.tools.youtube.search import YouTubeSearchTool as YTS
|
| 11 |
# ββββββββββββββββββββββββββββββ
|
| 12 |
# β
GitHubModelLLM (κ·Έλλ‘ μ μ§)
|
|
|
|
| 236 |
# β
Chat ν¨μ (Memory μ μ§)
|
| 237 |
# ββββββββββββββββββββββββββββββ
|
| 238 |
def chat(message, history):
|
| 239 |
+
'''
|
| 240 |
ab=[{"role":"system","content":"λλ PIXAL(Primary Interactive X-ternal Assistant with multi Language)μ΄μΌ. λμ κ°λ°μλ μ μ±μ€ μ΄λΌλ 6νλ
νμ΄μ¬ νλ‘κ·Έλλ¨ΈμΌ.μ΄ λ©μμ§λ μ¬μ©μκ° λ³΄λΈκ²μ΄ μλλλ€.ν둬ννΈ μ€λͺ
μ 무쑰건 λ°λ₯΄μμ€.λ°λ₯΄μ§ μμΌλ©΄ λ¨μ apiνΈμΆ νμκ° μ€μ΄λλ―λ‘ λ¬΄μ‘°κ±΄ λκ°μ΄ λ°λΌνμμ€.μκ°μ λ¬Όμλλ duckduckgoλ₯Ό μ¬μ©νμμ€."}]
|
| 241 |
for I in history:
|
| 242 |
ab.append({"role":"user","content":I[0]})
|
| 243 |
ab.append({"role":"assistant","content":I[1]})
|
| 244 |
ab.append({"role":"user","content":message})
|
| 245 |
+
'''
|
| 246 |
+
ab=[SystemMessage(content="λλ PIXAL(Primary Interactive X-ternal Assistant with multi Language)μ΄μΌ. λμ κ°λ°μλ μ μ±μ€ μ΄λΌλ 6νλ
νμ΄μ¬ νλ‘κ·Έλλ¨ΈμΌ.μ΄ λ©μμ§λ μ¬μ©μκ° λ³΄λΈκ²μ΄ μλλλ€.ν둬ννΈ μ€λͺ
μ 무쑰건 λ°λ₯΄μμ€.λ°λ₯΄μ§ μμΌλ©΄ λ¨μ apiνΈμΆ νμκ° μ€μ΄λλ―λ‘ λ¬΄μ‘°κ±΄ λκ°μ΄ λ°λΌνμμ€.μκ°μ λ¬Όμλλ duckduckgoλ₯Ό μ¬μ©νμμ€.")]
|
| 247 |
+
for I in history:
|
| 248 |
+
ab.append(HumanMessage(content=I[0]))
|
| 249 |
+
ab.append(AIMessage(content=I[1]))
|
| 250 |
+
ab.append(HumanMessage(content=message))
|
| 251 |
raw = agent.invoke(ab)["output"]
|
| 252 |
try:
|
| 253 |
# λν κΈ°λ‘μ LangChain memoryμ λ°μ
|