From 0a7392f6ec01b0701a1c7298252158d13db988ea Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Wed, 20 Mar 2024 17:03:08 +0530 Subject: [PATCH] Only add location to image prompt generator when location known --- src/khoj/routers/helpers.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/src/khoj/routers/helpers.py b/src/khoj/routers/helpers.py index 7a736663..5c3637b6 100644 --- a/src/khoj/routers/helpers.py +++ b/src/khoj/routers/helpers.py @@ -326,10 +326,13 @@ async def generate_better_image_prompt( Generate a better image prompt from the given query """ - location = f"{location_data.city}, {location_data.region}, {location_data.country}" if location_data else "Unknown" today_date = datetime.now(tz=timezone.utc).strftime("%Y-%m-%d") - location_prompt = prompts.user_location.format(location=location) + if location_data: + location = f"{location_data.city}, {location_data.region}, {location_data.country}" + location_prompt = prompts.user_location.format(location=location) + else: + location_prompt = "Unknown" user_references = "\n\n".join([f"# {item}" for item in note_references])