From 8a64d9c959baee020eaee9218dfb92a15a8e0d63 Mon Sep 17 00:00:00 2001 From: armistace Date: Mon, 19 May 2025 11:38:15 +1000 Subject: [PATCH] fix pyrefly typuing errors --- src/ai_generators/ollama_md_generator.py | 1 + src/trilium/notes.py | 4 ++++ 2 files changed, 5 insertions(+) diff --git a/src/ai_generators/ollama_md_generator.py b/src/ai_generators/ollama_md_generator.py index 4e140b0..99ac097 100644 --- a/src/ai_generators/ollama_md_generator.py +++ b/src/ai_generators/ollama_md_generator.py @@ -9,6 +9,7 @@ class OllamaGenerator: self.title = title self.inner_title = inner_title self.content = content + self.response = None self.chroma = chromadb.HttpClient(host="172.18.0.2", port=8000) ollama_url = f"{os.environ["OLLAMA_PROTOCOL"]}://{os.environ["OLLAMA_HOST"]}:{os.environ["OLLAMA_PORT"]}" self.ollama_client = Client(host=ollama_url) diff --git a/src/trilium/notes.py b/src/trilium/notes.py index bb55041..740fbfc 100644 --- a/src/trilium/notes.py +++ b/src/trilium/notes.py @@ -18,9 +18,13 @@ class TrilumNotes: print("Please run get_token and set your token") else: self.ea = ETAPI(self.server_url, self.token) + self.new_notes = None + self.note_content = None def get_token(self): ea = ETAPI(self.server_url) + if self.tril_pass == None: + raise ValueError("Trillium password can not be none") token = ea.login(self.tril_pass) print(token) print("I would recomend you update the env file with this tootsweet!")