From 96eed5fa493329777bec863376bf48f93f442471 Mon Sep 17 00:00:00 2001 From: Shreyas Pimpalgaonkar Date: Fri, 13 Sep 2024 10:14:04 -0700 Subject: [PATCH] minor fix --- py/core/providers/parsing/unstructured_parsing.py | 3 +++ py/core/providers/prompts/r2r_prompts.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/py/core/providers/parsing/unstructured_parsing.py b/py/core/providers/parsing/unstructured_parsing.py index ac5f2a862..e541661ce 100644 --- a/py/core/providers/parsing/unstructured_parsing.py +++ b/py/core/providers/parsing/unstructured_parsing.py @@ -217,6 +217,9 @@ async def parse( metadata=metadata, ) + if iteration == 0: + raise ValueError(f"No chunks found for document {document.id}") + logger.debug( f"Parsed document with id={document.id}, title={document.metadata.get('title', None)}, " f"user_id={document.metadata.get('user_id', None)}, metadata={document.metadata} " diff --git a/py/core/providers/prompts/r2r_prompts.py b/py/core/providers/prompts/r2r_prompts.py index a0276f6a7..e830a80f9 100644 --- a/py/core/providers/prompts/r2r_prompts.py +++ b/py/core/providers/prompts/r2r_prompts.py @@ -4,11 +4,11 @@ import os from pathlib import Path from typing import Any, Optional -from uuid import uuid5, NAMESPACE_DNS import yaml from core.base import Prompt, PromptConfig, PromptProvider, R2RException +from core.base.utils import generate_id_from_label from core.providers.database.postgres import PostgresDBProvider logger = logging.getLogger(__name__) @@ -218,7 +218,7 @@ async def _save_prompt_to_database(self, prompt: Prompt): await self.execute_query( query, [ - uuid4(), + generate_id_from_label(prompt.name), prompt.name, prompt.template, json.dumps(prompt.input_types),