Spaces:
Sleeping
Sleeping
fix: use custom tool template only for huggingchat
Browse files- src/lib/server/models.ts +2 -2
src/lib/server/models.ts
CHANGED
|
@@ -130,7 +130,7 @@ async function getChatPromptRender(
|
|
| 130 |
// or use the `rag` mode without the citations
|
| 131 |
const id = m.id ?? m.name;
|
| 132 |
|
| 133 |
-
if (id.startsWith("CohereForAI")) {
|
| 134 |
formattedMessages = [
|
| 135 |
{
|
| 136 |
role: "system",
|
|
@@ -157,7 +157,7 @@ async function getChatPromptRender(
|
|
| 157 |
},
|
| 158 |
...formattedMessages,
|
| 159 |
];
|
| 160 |
-
} else if (id.startsWith("meta-llama")) {
|
| 161 |
const results = toolResults.flatMap((result) => {
|
| 162 |
if (result.status === ToolResultStatus.Error) {
|
| 163 |
return [
|
|
|
|
| 130 |
// or use the `rag` mode without the citations
|
| 131 |
const id = m.id ?? m.name;
|
| 132 |
|
| 133 |
+
if (isHuggingChat && id.startsWith("CohereForAI")) {
|
| 134 |
formattedMessages = [
|
| 135 |
{
|
| 136 |
role: "system",
|
|
|
|
| 157 |
},
|
| 158 |
...formattedMessages,
|
| 159 |
];
|
| 160 |
+
} else if (isHuggingChat && id.startsWith("meta-llama")) {
|
| 161 |
const results = toolResults.flatMap((result) => {
|
| 162 |
if (result.status === ToolResultStatus.Error) {
|
| 163 |
return [
|