changes

app/actions/ollama.ts

@@ -0,0 +1,51 @@
+'use server';
+
+import ollama from 'ollama';
+
+export interface OllamaModel {
+  name: string;
+  size: number;
+  digest: string;
+  details: {
+    format: string;
+    family: string;
+    families: string[];
+    parameter_size: string;
+    quantization_level: string;
+  };
+}
+
+export async function getInstalledModels(): Promise<OllamaModel[]> {
+  try {
+    const response = await ollama.list();
+    return response.models as OllamaModel[];
+  } catch (error) {
+    console.error('Error fetching models:', error);
+    return [];
+  }
+}
+
+export async function pullModel(modelName: string): Promise<{ success: boolean; message: string }> {
+  try {
+    // This awaits the full pull. For large models, this might timeout the server action.
+    // Ideally we would stream this, but for now we'll try a simple await.
+    // Next.js Server Actions have a default timeout.
+    await ollama.pull({ model: modelName });
+    return { success: true, message: `Successfully pulled ${modelName}` };
+  } catch (error: any) {
+    console.error('Error pulling model:', error);
+    return { success: false, message: error.message || 'Failed to pull model' };
+  }
+}
+
+export async function deleteModel(
+  modelName: string
+): Promise<{ success: boolean; message: string }> {
+  try {
+    await ollama.delete({ model: modelName });
+    return { success: true, message: `Successfully deleted ${modelName}` };
+  } catch (error: any) {
+    console.error('Error deleting model:', error);
+    return { success: false, message: error.message || 'Failed to delete model' };
+  }
+}