fix(main): load args if no config file

Author: ReturnToFirst

main.py

@@ -1,16 +1,18 @@
 from fastapi import FastAPI, Query
 from fastapi.responses import PlainTextResponse
 import uvicorn
-from prompt import Prompt
+
 from config import Config, parse_args
 from openai_client import LLMClient
 from db import DB
 
 
 proxy_server = FastAPI()
 args = parse_args()
-if args.config_file:
-    config = Config.from_toml("config.toml")
+if args.config:
+    config = Config.from_toml(args.config)
+else:
+    config = Config.from_args(args)
 client = LLMClient.from_config(config)
 db = DB.from_file(config.database_config.db_file)
 

openai_client.py

@@ -51,6 +51,7 @@ def request_completion(self):
             str: The content of the response message from the language model.
             Exception: An exception if an error occurs during the API call.
         """
+
         try:
             completion = self.client.chat.completions.create(
                         model=self.config.openai_config.model_name,