VeuReu commited on
Commit
24f0e97
·
verified ·
1 Parent(s): 6ec9a3a

Update main_process/salamandra_router.py

Browse files
Files changed (1) hide show
  1. main_process/salamandra_router.py +2 -2
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'