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: