|
1 |
| -import { AzureOpenAiApiVersionsList, EnginesConfiguration, LLMProvider, OpenAIModelsList } from '../../../shared/model/ai-settings'; |
| 1 | +import { |
| 2 | + AzureOpenAiApiVersionsList, |
| 3 | + EnginesConfiguration, |
| 4 | + LLMProvider, |
| 5 | + OllamaLlmModelsList, |
| 6 | + OpenAIModelsList |
| 7 | +} from '../../../shared/model/ai-settings'; |
2 | 8 |
|
3 | 9 | export const DefaultPrompt: string = `# Sentences generation instructions
|
4 | 10 |
|
@@ -54,5 +60,15 @@ export const EngineConfigurations: EnginesConfiguration[] = [
|
54 | 60 | { key: 'temperature', label: 'Temperature', type: 'number', inputScale: 'fullwidth' },
|
55 | 61 | { key: 'prompt', label: 'Prompt', type: 'prompt', inputScale: 'fullwidth', defaultValue: DefaultPrompt }
|
56 | 62 | ]
|
| 63 | + }, |
| 64 | + { |
| 65 | + label: 'Ollama', |
| 66 | + key: LLMProvider.Ollama, |
| 67 | + params: [ |
| 68 | + { key: 'baseUrl', label: 'BaseUrl', type: 'text', defaultValue: 'http://localhost:11434' }, |
| 69 | + { key: 'model', label: 'Model', type: 'openlist', source: OllamaLlmModelsList, defaultValue: 'llama2' }, |
| 70 | + { key: 'temperature', label: 'Temperature', type: 'number', inputScale: 'fullwidth', defaultValue: 0.7 }, |
| 71 | + { key: 'prompt', label: 'Prompt', type: 'prompt', inputScale: 'fullwidth', defaultValue: DefaultPrompt } |
| 72 | + ] |
57 | 73 | }
|
58 | 74 | ];
|
0 commit comments