arshiaafshani commited on
Commit
b3f7144
·
verified ·
1 Parent(s): 8a1b0f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -3,6 +3,7 @@ from huggingface_hub import hf_hub_download
3
  from llama_cpp import Llama
4
  from llama_cpp_agent import LlamaCppAgent, MessagesFormatterType
5
  from llama_cpp_agent.providers import LlamaCppPythonProvider
 
6
  from llama_cpp_agent.chat_history.messages import UserMessage, AssistantMessage
7
 
8
  # ⬇️ دانلود مدل
@@ -40,16 +41,17 @@ def respond(message, chat_history, system_message, max_tokens, temperature, top_
40
  settings.repeat_penalty = repeat_penalty
41
  settings.stream = True
42
 
43
- # رفرش سابقه چت
44
  if chat_history is None:
45
  chat_history = []
46
 
47
- messages = []
 
 
48
  for msg in chat_history:
49
  if msg["role"] == "user":
50
- messages.append(UserMessage(msg["content"]))
51
  elif msg["role"] == "assistant":
52
- messages.append(AssistantMessage(msg["content"]))
53
 
54
  # گرفتن استریم پاسخ
55
  stream = agent.get_chat_response(
 
3
  from llama_cpp import Llama
4
  from llama_cpp_agent import LlamaCppAgent, MessagesFormatterType
5
  from llama_cpp_agent.providers import LlamaCppPythonProvider
6
+ from llama_cpp_agent.chat_history import BasicChatHistory
7
  from llama_cpp_agent.chat_history.messages import UserMessage, AssistantMessage
8
 
9
  # ⬇️ دانلود مدل
 
41
  settings.repeat_penalty = repeat_penalty
42
  settings.stream = True
43
 
 
44
  if chat_history is None:
45
  chat_history = []
46
 
47
+ # استفاده صحیح از BasicChatHistory
48
+ messages = BasicChatHistory()
49
+
50
  for msg in chat_history:
51
  if msg["role"] == "user":
52
+ messages.add_message(UserMessage(msg["content"]))
53
  elif msg["role"] == "assistant":
54
+ messages.add_message(AssistantMessage(msg["content"]))
55
 
56
  # گرفتن استریم پاسخ
57
  stream = agent.get_chat_response(