Update main_process/salamandra_router.py
Browse files
main_process/salamandra_router.py
CHANGED
|
@@ -66,7 +66,7 @@ class SalamandraClient:
|
|
| 66 |
use_auth_token= HF_TOKEN
|
| 67 |
)
|
| 68 |
|
| 69 |
-
def chat(self, message) -> str:
|
| 70 |
date_string = datetime.today().strftime('%Y-%m-%d')
|
| 71 |
|
| 72 |
prompt = self.tokenizer.apply_chat_template(
|
|
@@ -1351,7 +1351,7 @@ async def generate_salamadra_result(
|
|
| 1351 |
|
| 1352 |
messages_registro.append({"role": "user", "content": user_prompt})
|
| 1353 |
|
| 1354 |
-
response = salamandra_client.chat(messages)
|
| 1355 |
print(f"[Salamandra] {response}")
|
| 1356 |
|
| 1357 |
# Extraer lo que viene después de 'assistant'
|
|
|
|
| 66 |
use_auth_token= HF_TOKEN
|
| 67 |
)
|
| 68 |
|
| 69 |
+
def chat(self, message, tools) -> str:
|
| 70 |
date_string = datetime.today().strftime('%Y-%m-%d')
|
| 71 |
|
| 72 |
prompt = self.tokenizer.apply_chat_template(
|
|
|
|
| 1351 |
|
| 1352 |
messages_registro.append({"role": "user", "content": user_prompt})
|
| 1353 |
|
| 1354 |
+
response = salamandra_client.chat(messages,tools)
|
| 1355 |
print(f"[Salamandra] {response}")
|
| 1356 |
|
| 1357 |
# Extraer lo que viene después de 'assistant'
|