Update app.py
Browse files
app.py
CHANGED
@@ -5,15 +5,11 @@ from datetime import datetime
|
|
5 |
from typing import Literal
|
6 |
import os
|
7 |
import importlib
|
8 |
-
from llm_handler import send_to_llm
|
9 |
from main import generate_data, PROMPT_1
|
10 |
from topics import TOPICS
|
11 |
from system_messages import SYSTEM_MESSAGES_VODALUS
|
12 |
import random
|
13 |
-
import llamanet
|
14 |
-
|
15 |
-
# Initialize LlamaNet
|
16 |
-
llamanet.run()
|
17 |
|
18 |
ANNOTATION_CONFIG_FILE = "annotation_config.json"
|
19 |
OUTPUT_FILE_PATH = "dataset.jsonl"
|
@@ -276,7 +272,7 @@ def chat_with_llm(message, history):
|
|
276 |
msg_list.append({"role": "assistant", "content": h[1]})
|
277 |
msg_list.append({"role": "user", "content": message})
|
278 |
|
279 |
-
response, _ = send_to_llm(
|
280 |
|
281 |
return history + [[message, response]]
|
282 |
except Exception as e:
|
|
|
5 |
from typing import Literal
|
6 |
import os
|
7 |
import importlib
|
8 |
+
from llm_handler import send_to_llm, agent, settings
|
9 |
from main import generate_data, PROMPT_1
|
10 |
from topics import TOPICS
|
11 |
from system_messages import SYSTEM_MESSAGES_VODALUS
|
12 |
import random
|
|
|
|
|
|
|
|
|
13 |
|
14 |
ANNOTATION_CONFIG_FILE = "annotation_config.json"
|
15 |
OUTPUT_FILE_PATH = "dataset.jsonl"
|
|
|
272 |
msg_list.append({"role": "assistant", "content": h[1]})
|
273 |
msg_list.append({"role": "user", "content": message})
|
274 |
|
275 |
+
response, _ = send_to_llm(agent, msg_list)
|
276 |
|
277 |
return history + [[message, response]]
|
278 |
except Exception as e:
|