fix: allow api call in stream route + get_prompt error

This commit is contained in:
Siddhant Rai
2025-07-25 16:17:18 +05:30
parent 76973a4b4c
commit 212952f3e9
3 changed files with 3 additions and 4 deletions

View File

@@ -63,7 +63,8 @@ class AnswerResource(Resource, BaseAnswerResource):
data = request.get_json()
if error := self.validate_request(data):
return error
processor = StreamProcessor(data, None)
decoded_token = getattr(request, "decoded_token", None)
processor = StreamProcessor(data, decoded_token)
try:
processor.initialize()
if not processor.decoded_token:

View File

@@ -70,8 +70,6 @@ class StreamResource(Resource, BaseAnswerResource):
if error := self.validate_request(data, "index" in data):
return error
decoded_token = getattr(request, "decoded_token", None)
if not decoded_token:
return make_response({"error": "Unauthorized"}, 401)
processor = StreamProcessor(data, decoded_token)
try:
processor.initialize()

View File

@@ -41,7 +41,7 @@ def get_prompt(prompt_id: str, prompts_collection=None) -> str:
except FileNotFoundError:
raise FileNotFoundError(f"Prompt file not found: {file_path}")
try:
if not prompts_collection:
if prompts_collection is None:
mongo = MongoDB.get_client()
db = mongo[settings.MONGO_DB_NAME]
prompts_collection = db["prompts"]