quivr/frontend/lib/helpers/getAccessibleModels.ts
Antoine Dewez b5e2d5ad9c
feat(frontend): Add Brain On Search Page (#2067)
# Description

- Implement Icon Component
- Implement TextButton Component
- Change Add Brain Button And Set it in the Search Page
- Fix Errors When sending empty message
- Change EsLint rules

## 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):
2024-01-22 17:37:45 -08:00

21 lines
535 B
TypeScript

import { freeModels, paidModels } from "@/lib/types/BrainConfig";
import { UserStats } from "@/lib/types/User";
type GetAccessibleModelsInput = {
openAiKey?: string | null;
userData?: UserStats;
};
export const getAccessibleModels = ({
openAiKey,
userData,
}: GetAccessibleModelsInput): string[] => {
if (userData?.models !== undefined) {
return userData.models;
}
if (openAiKey !== undefined && openAiKey !== null) {
return paidModels as unknown as string[];
}
return freeModels as unknown as string[];
};