Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -172,9 +172,7 @@ class LlamaCppModel(Model):
|
|
| 172 |
**kwargs,
|
| 173 |
) -> ChatMessage:
|
| 174 |
|
| 175 |
-
|
| 176 |
-
Generates a response from the llama.cpp model and integrates tool usage *only if tools are provided*.
|
| 177 |
-
"""
|
| 178 |
|
| 179 |
from llama_cpp import LlamaGrammar
|
| 180 |
try:
|
|
@@ -183,7 +181,6 @@ class LlamaCppModel(Model):
|
|
| 183 |
stop_sequences=stop_sequences,
|
| 184 |
grammar=grammar,
|
| 185 |
tools_to_call_from=tools_to_call_from,
|
| 186 |
-
#flatten_messages_as_text=True,
|
| 187 |
**kwargs
|
| 188 |
)
|
| 189 |
|
|
@@ -201,7 +198,7 @@ class LlamaCppModel(Model):
|
|
| 201 |
or 1024
|
| 202 |
)
|
| 203 |
|
| 204 |
-
|
| 205 |
system_message= completion_kwargs["messages"][0]["content"]
|
| 206 |
message= completion_kwargs["messages"].pop()["content"]
|
| 207 |
|
|
@@ -225,7 +222,7 @@ class LlamaCppModel(Model):
|
|
| 225 |
settings.repeat_penalty = repeat_penalty
|
| 226 |
settings.stream = False
|
| 227 |
|
| 228 |
-
|
| 229 |
messages = BasicChatHistory()
|
| 230 |
for from_message in completion_kwargs["messages"]:
|
| 231 |
if from_message["role"] is MessageRole.USER:
|
|
@@ -235,7 +232,7 @@ class LlamaCppModel(Model):
|
|
| 235 |
else:
|
| 236 |
history_message = {"role": MessageRole.ASSISTANT, "content": from_message["content"]}
|
| 237 |
messages.add_message(from_message)
|
| 238 |
-
|
| 239 |
stream = agent.get_chat_response(
|
| 240 |
message,
|
| 241 |
llm_sampling_settings=settings,
|
|
|
|
| 172 |
**kwargs,
|
| 173 |
) -> ChatMessage:
|
| 174 |
|
| 175 |
+
|
|
|
|
|
|
|
| 176 |
|
| 177 |
from llama_cpp import LlamaGrammar
|
| 178 |
try:
|
|
|
|
| 181 |
stop_sequences=stop_sequences,
|
| 182 |
grammar=grammar,
|
| 183 |
tools_to_call_from=tools_to_call_from,
|
|
|
|
| 184 |
**kwargs
|
| 185 |
)
|
| 186 |
|
|
|
|
| 198 |
or 1024
|
| 199 |
)
|
| 200 |
|
| 201 |
+
provider = LlamaCppPythonProvider(self.llm)
|
| 202 |
system_message= completion_kwargs["messages"][0]["content"]
|
| 203 |
message= completion_kwargs["messages"].pop()["content"]
|
| 204 |
|
|
|
|
| 222 |
settings.repeat_penalty = repeat_penalty
|
| 223 |
settings.stream = False
|
| 224 |
|
| 225 |
+
|
| 226 |
messages = BasicChatHistory()
|
| 227 |
for from_message in completion_kwargs["messages"]:
|
| 228 |
if from_message["role"] is MessageRole.USER:
|
|
|
|
| 232 |
else:
|
| 233 |
history_message = {"role": MessageRole.ASSISTANT, "content": from_message["content"]}
|
| 234 |
messages.add_message(from_message)
|
| 235 |
+
|
| 236 |
stream = agent.get_chat_response(
|
| 237 |
message,
|
| 238 |
llm_sampling_settings=settings,
|