From bf40e4a3ca9c88a5d9e925a169a2acb3d204f325 Mon Sep 17 00:00:00 2001 From: Andrew Ridgway Date: Mon, 11 Nov 2024 20:23:18 +1000 Subject: [PATCH] set up for proper repo crap --- .gitignore | 3 +++ Dockerfile | 2 +- requirements.txt | 2 +- src/main.py | 13 +++++++++++++ src/test.py | 1 - src/trilium/__init__.py | 0 src/trilium/get_token.py | 8 ++++++++ 7 files changed, 26 insertions(+), 3 deletions(-) create mode 100644 .gitignore create mode 100644 src/main.py delete mode 100644 src/test.py create mode 100644 src/trilium/__init__.py create mode 100644 src/trilium/get_token.py diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..5152f84 --- /dev/null +++ b/.gitignore @@ -0,0 +1,3 @@ +.env +__pycache__ +.venv diff --git a/Dockerfile b/Dockerfile index 7e58aa9..cd9bfda 100644 --- a/Dockerfile +++ b/Dockerfile @@ -12,4 +12,4 @@ RUN pip install -r requirements.txt #RUN make html -ENTRYPOINT ["python", "test.py"] +ENTRYPOINT ["python", "main.py"] diff --git a/requirements.txt b/requirements.txt index fb6c7ed..403abba 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1 @@ -pandas +ollama diff --git a/src/main.py b/src/main.py new file mode 100644 index 0000000..d7f7ca9 --- /dev/null +++ b/src/main.py @@ -0,0 +1,13 @@ +from ollama import Client + +client = Client(host='http://192.168.178.45:11434') + +user_prompt = input("Ask mistral-nemo a question: ") + +response = client.chat(model='mistral-nemo', messages=[ + { + 'role': 'user', + 'content': f'{user_prompt}', + }, +]) +print(response['message']['content']) diff --git a/src/test.py b/src/test.py deleted file mode 100644 index 2162131..0000000 --- a/src/test.py +++ /dev/null @@ -1 +0,0 @@ -print("This is working") diff --git a/src/trilium/__init__.py b/src/trilium/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/trilium/get_token.py b/src/trilium/get_token.py new file mode 100644 index 0000000..707136f --- /dev/null +++ b/src/trilium/get_token.py @@ -0,0 +1,8 @@ +from trilium_py.client import ETAPI +import os + +server_url = f'{os.environ['TRILIUM_PROTOCOL']}://{os.environ['TRILIUM_HOST']}:{os.environ['TRILIUM_PORT']}' +password = os.environ['TRILIUM_PASS'] +ea = ETAPI(server_url) +token = ea.login(password) +print(token)