|
18 | 18 | import com.developer.nefarious.zjoule.plugin.models.OllamaModel;
|
19 | 19 |
|
20 | 20 | public class SecondOllamaLoginWizardPage extends WizardPage {
|
21 |
| - |
| 21 | + |
22 | 22 | public static final String PAGE_ID = "Ollama Login Second Page";
|
23 |
| - |
| 23 | + |
24 | 24 | private Combo ollamaModelDropdown;
|
25 |
| - |
| 25 | + |
26 | 26 | private List<OllamaModel> ollamaModelsForSelection = new ArrayList<>();
|
27 |
| - |
| 27 | + |
28 | 28 | private IMemoryObject<OllamaModel> memoryOllamaModel;
|
29 |
| - |
| 29 | + |
30 | 30 | public SecondOllamaLoginWizardPage(final IMemoryObject<OllamaModel> memoryOllamaModel) {
|
31 | 31 | super(PAGE_ID);
|
32 | 32 | setTitle("Ollama Setup");
|
33 |
| - setDescription("Select the Ollama model."); |
34 |
| - setPageComplete(false); // Initially set the page as incomplete |
35 |
| - this.memoryOllamaModel = memoryOllamaModel; |
| 33 | + setDescription("Select the Ollama model."); |
| 34 | + setPageComplete(false); // Initially set the page as incomplete |
| 35 | + this.memoryOllamaModel = memoryOllamaModel; |
36 | 36 | }
|
37 | 37 |
|
38 | 38 | @Override
|
39 | 39 | public void createControl(final Composite parent) {
|
40 | 40 | Composite container = new Composite(parent, SWT.NONE);
|
41 | 41 | container.setLayout(new GridLayout(2, false));
|
42 |
| - |
43 |
| - Label modelLabel = new Label(container, SWT.NONE); |
44 |
| - modelLabel.setText("Select the Model:"); |
45 |
| - |
46 |
| - ollamaModelDropdown = new Combo(container, SWT.DROP_DOWN | SWT.READ_ONLY); |
47 |
| - ollamaModelDropdown.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false)); |
48 |
| - |
49 |
| - ollamaModelDropdown.addSelectionListener(new OllamaModelSelectionAdapter(this, memoryOllamaModel)); |
50 |
| - |
51 |
| - setControl(container); |
| 42 | + |
| 43 | + Label modelLabel = new Label(container, SWT.NONE); |
| 44 | + modelLabel.setText("Select the Model:"); |
| 45 | + |
| 46 | + ollamaModelDropdown = new Combo(container, SWT.DROP_DOWN | SWT.READ_ONLY); |
| 47 | + ollamaModelDropdown.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false)); |
| 48 | + |
| 49 | + ollamaModelDropdown.addSelectionListener(new OllamaModelSelectionAdapter(this, memoryOllamaModel)); |
| 50 | + |
| 51 | + setControl(container); |
52 | 52 | }
|
53 |
| - |
| 53 | + |
54 | 54 | public List<OllamaModel> getOllamaModelsForSelection() {
|
55 |
| - return ollamaModelsForSelection; |
56 |
| - } |
57 |
| - |
| 55 | + return ollamaModelsForSelection; |
| 56 | + } |
| 57 | + |
58 | 58 | public void setOllamaModelsForSelection(final GetOllamaModelsResponse getOllamaModelsResponse) {
|
59 | 59 | this.ollamaModelsForSelection = getOllamaModelsResponse.getModels();
|
60 | 60 | List<String> ollamaModelNames = OllamaModelNamesExtractor.extractModelNames(getOllamaModelsResponse);
|
61 | 61 | ollamaModelDropdown.setItems(ollamaModelNames.toArray(new String[0]));
|
62 | 62 | }
|
63 |
| - |
| 63 | + |
64 | 64 | public Combo getOllamaModelDropdown() {
|
65 |
| - return ollamaModelDropdown; |
66 |
| - } |
| 65 | + return ollamaModelDropdown; |
| 66 | + } |
| 67 | + |
| 68 | + @Override |
| 69 | + public void setVisible(final boolean visible) { |
| 70 | + super.setVisible(visible); |
| 71 | + if (!visible) { |
| 72 | + ollamaModelDropdown.deselectAll(); |
| 73 | + setPageComplete(false); |
| 74 | + } |
| 75 | + } |
67 | 76 |
|
68 | 77 | }
|
0 commit comments