Update main.py
Browse files
main.py
CHANGED
@@ -1,20 +1,12 @@
|
|
1 |
# Import necessary libraries and modules
|
2 |
-
import json
|
3 |
-
import numpy as np
|
4 |
-
from wiki import search as search_wikipedia
|
5 |
-
from concurrent.futures import ThreadPoolExecutor
|
6 |
-
from llm_handler import send_to_llm
|
7 |
-
from params import OUTPUT_FILE_PATH, NUM_WORKERS
|
8 |
-
import
|
9 |
-
|
10 |
-
# Import system messages from the system_messages module
|
11 |
-
from system_messages import (
|
12 |
-
SYSTEM_MESSAGES_VODALUS,
|
13 |
-
)
|
14 |
-
from topics import TOPICS # Import topics from the topics module
|
15 |
-
|
16 |
-
# Initialize LlamaNet
|
17 |
-
llamanet.run()
|
18 |
|
19 |
# Set the system messages to those specified in SYSTEM_MESSAGES_VODALUS
|
20 |
SYSTEM_MESSAGES = SYSTEM_MESSAGES_VODALUS
|
@@ -118,6 +110,5 @@ def main():
|
|
118 |
failed += 1
|
119 |
print("=" * 132)
|
120 |
|
121 |
-
|
122 |
if __name__ == "__main__":
|
123 |
-
main()
|
|
|
1 |
# Import necessary libraries and modules
|
2 |
+
import json
|
3 |
+
import numpy as np
|
4 |
+
from wiki import search as search_wikipedia
|
5 |
+
from concurrent.futures import ThreadPoolExecutor
|
6 |
+
from llm_handler import send_to_llm
|
7 |
+
from params import OUTPUT_FILE_PATH, NUM_WORKERS
|
8 |
+
from system_messages import SYSTEM_MESSAGES_VODALUS
|
9 |
+
from topics import TOPICS
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
10 |
|
11 |
# Set the system messages to those specified in SYSTEM_MESSAGES_VODALUS
|
12 |
SYSTEM_MESSAGES = SYSTEM_MESSAGES_VODALUS
|
|
|
110 |
failed += 1
|
111 |
print("=" * 132)
|
112 |
|
|
|
113 |
if __name__ == "__main__":
|
114 |
+
main()
|