quivr/backend/routes/chat/factory.py
Zineb El Bachiri e3a99d1ace
refactor: chat_routes (#1512)
# 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.

- [ ] My code follows the style guidelines of this project
- [ ] 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):
2023-10-30 10:18:23 +01:00

12 lines
248 B
Python

from uuid import UUID
from .brainful_chat import BrainfulChat
from .brainless_chat import BrainlessChat
def get_chat_strategy(brain_id: UUID | None = None):
if brain_id:
return BrainfulChat()
else:
return BrainlessChat()