[FIX] Fixed issue where context is empty (#1211)

Deepak-Kesavan · ritwik-g · web-flow · commit d8810cf5627d · 2025-03-27T18:55:11.000+05:30
Added the context if found

Co-authored-by: Ritwik G &lt;100672805+ritwik-g@users.noreply.github.com&gt;
diff --git a/prompt-service/src/unstract/prompt_service/main.py b/prompt-service/src/unstract/prompt_service/main.py
@@ -782,6 +782,7 @@ def fetch_context_from_vector_db(
                     msg,
                 )
                 raise APIError(message=msg)
+            context.add(retrieved_context)
     except SdkError as e:
         msg = f"Unable to fetch context from vector DB. {str(e)}"
         app.logger.error(

Original file line number	Diff line number	Diff line change
`@@ -782,6 +782,7 @@ def fetch_context_from_vector_db(`
`782`	`782`	`msg,`
`783`	`783`	`)`
`784`	`784`	`raise APIError(message=msg)`
	`785`	`+ context.add(retrieved_context)`
`785`	`786`	`except SdkError as e:`
`786`	`787`	`msg = f"Unable to fetch context from vector DB. {str(e)}"`
`787`	`788`	`app.logger.error(`