quivr/backend/repository/brain/get_brain_prompt_id.py

11 lines
253 B
Python

from uuid import UUID
from repository.brain.get_brain_by_id import get_brain_by_id
def get_brain_prompt_id(brain_id: UUID) -> UUID | None:
brain = get_brain_by_id(brain_id)
prompt_id = brain.prompt_id if brain else None
return prompt_id