quivr/backend/modules/prompt/service/get_prompt_to_use.py
Thoonsen Maxime 5c965b6d22
refactor: reorg the files #2 (#2457)
# Description

Please include a summary of the changes and the related issue. Please
also include relevant motivation and context.

## Checklist before requesting a review

Please delete options that are not relevant.

- [x] My code follows the style guidelines of this project
- [x] I have performed a self-review of my code
- [ ] I have commented hard-to-understand areas
- [ ] I have ideally added tests that prove my fix is effective or that
my feature works
- [ ] New and existing unit tests pass locally with my changes
- [ ] Any dependent changes have been merged

## Screenshots (if appropriate):
2024-04-20 02:28:20 -07:00

16 lines
479 B
Python

from typing import Optional
from uuid import UUID
from modules.brain.service.utils.get_prompt_to_use_id import get_prompt_to_use_id
from modules.prompt.service import PromptService
promptService = PromptService()
def get_prompt_to_use(brain_id: Optional[UUID], prompt_id: Optional[UUID]) -> str:
prompt_to_use_id = get_prompt_to_use_id(brain_id, prompt_id)
if prompt_to_use_id is None:
return None
return promptService.get_prompt_by_id(prompt_to_use_id)