chore: set available models for ollama endpoint

Author: nickchecan

com.developer.nefarious.zjoule.plugin/src/com/developer/nefarious/zjoule/plugin/login/OllamaLoginWizard.java

@@ -3,28 +3,35 @@
 import org.eclipse.jface.wizard.Wizard;
 import org.eclipse.swt.browser.Browser;
 
+import com.developer.nefarious.zjoule.plugin.login.api.IOllamaLoginClient;
+import com.developer.nefarious.zjoule.plugin.login.api.OllamaLoginClient;
+import com.developer.nefarious.zjoule.plugin.login.api.OllamaLoginClientHelper;
+import com.developer.nefarious.zjoule.plugin.login.memory.TemporaryMemoryOllamaEndpoint;
 import com.developer.nefarious.zjoule.plugin.login.pages.FirstOllamaLoginPage;
 import com.developer.nefarious.zjoule.plugin.login.pages.SecondOllamaLoginPage;
 
 public class OllamaLoginWizard extends Wizard {
 
 	private Browser browser;
 
+	private IOllamaLoginClient ollamaLoginClient;
+	
 	public OllamaLoginWizard(final Browser browser) {
         this.browser = browser;
 
         setWindowTitle("Login to Ollama");
+        ollamaLoginClient = new OllamaLoginClient(new OllamaLoginClientHelper());
     }
 
 	@Override
     public void addPages() {
-        addPage(new FirstOllamaLoginPage());
+        addPage(new FirstOllamaLoginPage(ollamaLoginClient));
         addPage(new SecondOllamaLoginPage());
     }
 
 	@Override
 	public boolean performFinish() {
-//		TemporaryMemoryOllamaConfig.getInstance().persist();
+		TemporaryMemoryOllamaEndpoint.getInstance().persist();
 
 //		SessionManager.login(browser);
 		return true;

com.developer.nefarious.zjoule.plugin/src/com/developer/nefarious/zjoule/plugin/login/pages/FirstOllamaLoginPage.java

@@ -1,5 +1,8 @@
 package com.developer.nefarious.zjoule.plugin.login.pages;
 
+import java.io.IOException;
+import java.util.List;
+
 import org.eclipse.jface.wizard.IWizardPage;
 import org.eclipse.jface.wizard.WizardPage;
 import org.eclipse.swt.SWT;
@@ -9,20 +12,27 @@
 import org.eclipse.swt.widgets.Label;
 import org.eclipse.swt.widgets.Text;
 
+import com.developer.nefarious.zjoule.plugin.login.api.GetOllamaModelsResponse;
+import com.developer.nefarious.zjoule.plugin.login.api.IOllamaLoginClient;
+import com.developer.nefarious.zjoule.plugin.login.utils.OllamaModelNamesExtractor;
+
 public class FirstOllamaLoginPage extends WizardPage {
 
 	public static final String PAGE_ID = "Ollama Login First Page";
 
+	private IOllamaLoginClient ollamaLoginClient;
+
 	private Text endpointText;
-	
+
 	private Text errorText;
 
-	public FirstOllamaLoginPage() {
+	public FirstOllamaLoginPage(final IOllamaLoginClient ollamaLoginClient) {
 		super(PAGE_ID);
 
 		setTitle("Ollama Setup");
 		setDescription("Enter the host and port for the local Ollama instance.");
 		setPageComplete(false); // Initially set the page as incomplete
+		this.ollamaLoginClient = ollamaLoginClient;
 	}
 
 	@Override
@@ -36,39 +46,53 @@ public void createControl(final Composite parent) {
 		endpointText = new Text(container, SWT.BORDER);
 		endpointText.setLayoutData(new GridData(SWT.FILL, SWT.CENTER, true, false));
 		endpointText.setText("http://localhost:11434");
-		
+
 		// Hidden error text widget
-        errorText = new Text(container, SWT.READ_ONLY | SWT.MULTI | SWT.WRAP);
-        GridData errorTextGridData = new GridData(SWT.FILL, SWT.CENTER, true, false, 2, 1);
-        errorText.setLayoutData(errorTextGridData);
-        errorText.setForeground(container.getDisplay().getSystemColor(SWT.COLOR_RED));
-        errorText.setVisible(false); // Initially hidden
+		errorText = new Text(container, SWT.READ_ONLY | SWT.MULTI | SWT.WRAP);
+		GridData errorTextGridData = new GridData(SWT.FILL, SWT.CENTER, true, false, 2, 1);
+		errorText.setLayoutData(errorTextGridData);
+		errorText.setForeground(container.getDisplay().getSystemColor(SWT.COLOR_RED));
+		errorText.setVisible(false); // Initially hidden
 
 		setControl(container);
 	}
-	
+
 	@Override
-    public boolean canFlipToNextPage() {
-        return true;
-    }
+	public boolean canFlipToNextPage() {
+		return true;
+	}
 
 	@Override
 	public IWizardPage getNextPage() {
-		
-		String input = endpointText.getText();
-		
-		if (input == null || input.isEmpty() || input.isBlank()) {
+
+		String ollamaEndpoint = endpointText.getText();
+
+		if (ollamaEndpoint == null || ollamaEndpoint.isEmpty() || ollamaEndpoint.isBlank()) {
 			displayErrorMessage("Please, enter a local Ollama endpoint to proceed.");
 			return null;
 		}
-		
+
+		try {
+			setAvailableModelsFor(ollamaEndpoint);
+		} catch (IllegalArgumentException | IOException | InterruptedException e) {
+			displayErrorMessage("Local instance of Ollama invalid or doesn't exist in the informed address.");
+			return null;
+		}
+
 		errorText.setVisible(false);
 		return super.getNextPage(); // Proceed to the next page
 	}
-	
+
+	private void setAvailableModelsFor(final String ollamaEndpoint) throws IOException, InterruptedException {
+		GetOllamaModelsResponse response = ollamaLoginClient.getModels(ollamaEndpoint);
+		SecondOllamaLoginPage secondPage = (SecondOllamaLoginPage) getWizard().getPage(SecondOllamaLoginPage.PAGE_ID);
+		List<String> modelsAvailableForSelection = OllamaModelNamesExtractor.extractModelNames(response);
+		secondPage.setModelsForSelection(modelsAvailableForSelection);
+	}
+
 	private void displayErrorMessage(final String message) {
 		errorText.setText(message);
-        errorText.setVisible(true);
+		errorText.setVisible(true);
 	}
 
 }

com.developer.nefarious.zjoule.plugin/src/com/developer/nefarious/zjoule/plugin/login/utils/OllamaModelNamesExtractor.java

@@ -0,0 +1,38 @@
+package com.developer.nefarious.zjoule.plugin.login.utils;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import com.developer.nefarious.zjoule.plugin.login.api.GetOllamaModelsResponse;
+import com.developer.nefarious.zjoule.plugin.models.OllamaModel;
+
+public abstract class OllamaModelNamesExtractor {
+	
+	public static List<String> extractModelNames(final GetOllamaModelsResponse response) {
+        // Step 0: Check if response is null
+        if (response == null) {
+            return null;
+        }
+
+        // Step 1: Get the list of Model objects from the response
+        List<OllamaModel> ollamaModels = response.getModels();
+
+        // Step 2: Create an ArrayList to store the model names
+        List<String> ollamaModelNames = new ArrayList<>();
+
+        // Step 3 & 4: Loop through each Model and extract its name
+        for (OllamaModel ollamaModel : ollamaModels) {
+            String ollamaModelName = ollamaModel.getName();
+            ollamaModelNames.add(ollamaModelName);
+        }
+
+        // Return the list of model names
+        return ollamaModelNames;
+    }
+
+    /**
+     * Private constructor to prevent instantiation of this utility class.
+     */
+    private OllamaModelNamesExtractor() { }
+
+}