ArslanFOX commited on
Commit
f2387de
·
verified ·
1 Parent(s): 5d88938

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -19
app.py CHANGED
@@ -29,7 +29,7 @@ def final_answer(answer: str) -> str:
29
 
30
  try:
31
  model = HfApiModel(
32
- max_tokens=1000,
33
  temperature=0.5,
34
  model_id='mistralai/Mixtral-8x7B-Instruct-v0.1',
35
  )
@@ -38,27 +38,13 @@ except Exception as e:
38
 
39
  prompt_templates = {
40
  "system_prompt": (
41
- "Ты полезный помощник, который ВСЕГДА отвечает только на русском языке, без единого слова на английском. "
42
  "Для вопросов о времени (например, 'Сколько времени в [место]?') используй инструмент `get_current_time_in_timezone` "
43
  "с правильным часовым поясом (например, 'Europe/Moscow' для Москвы, 'Asia/Yekaterinburg' для Бирска). "
44
- "НИКОГДА не переопределяй инструменты используй только предоставленные. "
45
- "Всегда возвращай ответ через вызов `final_answer()`, а не через print или переменные. "
46
- "Если запрос неясен, уточни, но оставайся на русском."
47
  ),
48
- "default": "Ответ: {{question}}",
49
- "planning": {
50
- "initial_plan": (
51
- "Проанализируй запрос: {{question}}. Определи, требуется ли инструмент. "
52
- "Если запрос о времени, используй `get_current_time_in_timezone` с часовым поясом (например, 'Asia/Yekaterinburg' для Бирска). "
53
- "Опиши шаги и напиши код. Всегда возвращай результат через `final_answer()`."
54
- )
55
- },
56
- "managed_agent": (
57
- "Выполни задачу: {{question}}. Используй инструменты, если указано в плане. "
58
- "Код должен быть корректным и использовать только предоставленные инструменты. "
59
- "Верни результат через `final_answer()`."
60
- ),
61
- "final_answer": "Окончательный ответ: {{answer}}"
62
  }
63
 
64
  agent = CodeAgent(
 
29
 
30
  try:
31
  model = HfApiModel(
32
+ max_tokens=500,
33
  temperature=0.5,
34
  model_id='mistralai/Mixtral-8x7B-Instruct-v0.1',
35
  )
 
38
 
39
  prompt_templates = {
40
  "system_prompt": (
41
+ "Ты полезный помощник, который всегда отвечает на русском языке. "
42
  "Для вопросов о времени (например, 'Сколько времени в [место]?') используй инструмент `get_current_time_in_timezone` "
43
  "с правильным часовым поясом (например, 'Europe/Moscow' для Москвы, 'Asia/Yekaterinburg' для Бирска). "
44
+ "Не переопределяй инструменты, используй только те, что предоставлены. "
45
+ "Всегда возвращай ответ через вызов `final_answer()`, а не через print или переменные."
 
46
  ),
47
+ "default": "Ответ: {{question}}"
 
 
 
 
 
 
 
 
 
 
 
 
 
48
  }
49
 
50
  agent = CodeAgent(