fix pyrefly typuing errors

This commit is contained in:
armistace 2025-05-19 11:38:15 +10:00
parent 0c090c8489
commit 8a64d9c959
2 changed files with 5 additions and 0 deletions
src
ai_generators
trilium

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

@ -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!")