Add Space configuration and server files

.gitattributes

@@ -0,0 +1,35 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

index.js

@@ -0,0 +1,83 @@
+import express from 'express';
+import cors from 'cors';
+import axios from 'axios';
+import dotenv from 'dotenv';
+import { fileURLToPath } from 'url';
+import { dirname } from 'path';
+
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+
+dotenv.config();
+
+const app = express();
+
+// Enhanced CORS configuration
+app.use(cors({
+  origin: '*',
+  methods: ['GET', 'POST'],
+  allowedHeaders: ['Content-Type']
+}));
+
+app.use(express.json());
+
+const OLLAMA_URL = process.env.OLLAMA_URL || 'http://localhost:7860';
+
+// Model-specific configurations
+const MODEL_CONFIGS = {
+  'mxbai-embed-large': { temperature: 0.8, top_p: 0.9 },
+  'minicpm-v': { temperature: 0.8, top_p: 0.9 },
+  'qwen2.5-coder': { temperature: 0.6, top_p: 0.95 },
+  'codegemma': { temperature: 0.6, top_p: 0.95 },
+  'codellama': { temperature: 0.7, top_p: 0.95 },
+  'llama3.2-vision': { temperature: 0.8, top_p: 0.9 }
+};
+
+// Add these right after your app.use() statements
+app.get('/health', (req, res) => {
+  res.status(200).json({
+    status: 'healthy',
+    ollama: 'running',
+    model: 'minicpm-v'
+  });
+});
+
+// Add a root endpoint
+app.get('/', (req, res) => {
+  res.status(200).json({
+    message: 'Aide AI Backend Service',
+    version: '1.0'
+  });
+});
+
+// Update the chat endpoin`
+app.post('/api/chat', async (req, res) => {
+  const { message } = req.body;
+  try {
+    const response = await axios.post(`${OLLAMA_URL}/api/generate`, {
+      model: 'minicpm-v',
+      prompt: message,
+      stream: false,
+      options: {
+        num_ctx: 2048,
+        num_thread: 4
+      }
+    });
+    res.json({ response: response.data.response });
+  } catch (error) {
+    console.error('Error:', error);
+    res.status(500).json({ error: 'Model processing error' });
+  }
+});
+
+app.get('/api/models', async (req, res) => {
+  try {
+    const response = await axios.get(`${OLLAMA_URL}/api/tags`);
+    res.json(response.data);
+  } catch (error) {
+    res.status(500).json({ error: 'Failed to fetch models' });
+  }
+});
+
+const PORT = process.env.PORT || 7860;
+app.listen(PORT, '0.0.0.0', () => console.log(`Server running on port ${PORT}`));