From 8655a79e8b74adab8c3b001a7d37907c0de288ed Mon Sep 17 00:00:00 2001 From: Keannu Bernasol Date: Thu, 19 Dec 2024 00:12:11 +0800 Subject: [PATCH] Update prompt --- docmanager_backend/config/management/commands/start_watcher.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docmanager_backend/config/management/commands/start_watcher.py b/docmanager_backend/config/management/commands/start_watcher.py index fc456f9..a5dc38f 100644 --- a/docmanager_backend/config/management/commands/start_watcher.py +++ b/docmanager_backend/config/management/commands/start_watcher.py @@ -105,7 +105,7 @@ class PDFHandler(FileSystemEventHandler): prompt = f""" Read the text from the image and provide a category. Return as JSON. - Possible categories are: {possible_categories} + Possible categories are: {possible_categories}. You are free to create a new one if none are suitable. """ response = client.chat( model=get_secret("OLLAMA_MODEL"),