-Make server working again;

-Commented out arguments.py file and hardcoded constants for now;
-Resolve issue with bad naming and location of the db_get_incomplete_completion_tasks function

arguments.py

@@ -1,31 +1,32 @@
-import argparse
+# import argparse
 
-parser = argparse.ArgumentParser()
-parser.add_argument(
-    "-H",
-    "--use-hugging-face",
-    dest="use_hugging_face",
-    action="store_true",
-    help="Use Hugging Face as the model provider",
-)
-parser.add_argument(
-    "-M",
-    "--pick-model",
-    type=str,
-    dest="model_choice",
-    choices=[
-        "ollama_medium",
-        "ollama_small",
-        "ollama_large",
-        "hugging_face_medium",
-        "hugging_face_small",
-        "hugging_face_large",
-    ],
-    default="ollama_medium",
-    help="Select model configuration",
-)
+# parser = argparse.ArgumentParser()
+# parser.add_argument(
+#     "-H",
+#     "--use-hugging-face",
+#     dest="use_hugging_face",
+#     action="store_true",
+#     help="Use Hugging Face as the model provider",
+# )
+# parser.add_argument(
+#     "-M",
+#     "--pick-model",
+#     type=str,
+#     dest="model_choice",
+#     choices=[
+#         "ollama_medium",
+#         "ollama_small",
+#         "ollama_large",
+#         "hugging_face_medium",
+#         "hugging_face_small",
+#         "hugging_face_large",
+#     ],
+#     default="ollama_medium",
+#     help="Select model configuration",
+# )
 
-args = parser.parse_args()
+# args = parser.parse_args()
 
-USE_HUGGING_FACE = args.use_hugging_face
-MODEL_CHOICE = args.model_choice
+USE_HUGGING_FACE = True
+MODEL_CHOICE = "ollama_medium"
+#TODO: Handle the arguments parsing

webui/models/url_pool.py

@@ -1,5 +1,4 @@
 from __future__ import annotations
-
 from pydantic import BaseModel
 
 

webui/router/completion_routes.py

@@ -2,7 +2,7 @@
 from core.databases.db_completion_tasks import (
     db_add_completion_task,
     db_get_completion_tasks_by_page,
-    db_get_incomplete_completion_task,
+    db_get_incomplete_completion_tasks
 )
 from webui.models.completion_task import TaskCreator
 
@@ -17,7 +17,7 @@ def get_tasks_by_page():
 
 @router.get("/task/incomplete")
 def get_incomplete_task():
-    result = db_get_incomplete_completion_task()
+    result = db_get_incomplete_completion_tasks()
     return {"tasks": result}
 
 

webui/router/crawl_routes.py

@@ -3,8 +3,8 @@
     db_add_crawl_task,
     db_get_crawl_task,
     db_set_crawl_completed,
-    db_get_incomplete_completion_task,
 )
+from core.databases.db_completion_tasks import db_get_incomplete_completion_tasks
 from webui.models.crawl_task import CrawlCreator
 
 router = APIRouter()
@@ -29,5 +29,5 @@ def set_crawl_completed(uuid):
 
 @router.get("/crawl/incomplete")
 def get_inocmplete_completion_task():
-    result = db_get_incomplete_completion_task()
+    result = db_get_incomplete_completion_tasks()
     return {"task": result}