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()
|