diff --git a/src/Ai/llm.py b/src/Ai/llm.py index bf06849..f2a566c 100644 --- a/src/Ai/llm.py +++ b/src/Ai/llm.py @@ -1,5 +1,5 @@ from langchain_community.llms import Ollama - +import os class ArticleRater: def __init__(self): @@ -8,8 +8,12 @@ class ArticleRater: self.headers = {"Authorization": f"Token {self.token}"} def _get_token(self): - with open("Token/Token.txt", "r") as t: - return t.readline().strip() + if os.path.exists("Token/Token.txt"): + with open("Token/Token.txt", "r") as t: + return t.readline().strip() + else: + return None + def get_response(self, article, result, confidence): ollama_params = { "base_url": self.client,