From 74f912e7e330d8f07d3b411360a7ba3050b0da96 Mon Sep 17 00:00:00 2001 From: "l.gabrysiak" Date: Fri, 28 Feb 2025 21:15:21 +0100 Subject: [PATCH] mod allegro --- allegro.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/allegro.py b/allegro.py index 48a053d..80c24a2 100644 --- a/allegro.py +++ b/allegro.py @@ -6,6 +6,7 @@ from datasets import Dataset from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, TrainingArguments, Trainer, DataCollatorForSeq2Seq from weaviate.connect import ConnectionParams import weaviate +import tempfile # 1️⃣ Połączenie z Weaviate client = weaviate.WeaviateClient( @@ -25,7 +26,7 @@ def fetch_documents(): collection = client.collections.get("Document") response = collection.query.fetch_objects() documents = [] - for o in response.objects: + for o in response.objects.TemporaryDirectory(): file_name = o.properties.get("fileName", "unknown_file") content = o.properties.get("content", "") if content: