MLGraph API Documentation

import requests

# Initialize client
base_url = "http://localhost:8080/api/v1"
headers = {
    "Authorization": f"Bearer {token}"
}

# Create index
response = requests.post(
    f"{base_url}/indices",
    headers=headers,
    json={
        "index_id": "my-index",
        "dimension": 768,
        "index_type": "IVFFlat",
        "num_centroids": 100
    }
)

# Add vectors
vectors = generate_vectors(1000, 768)
requests.post(
    f"{base_url}/vectors",
    headers=headers,
    json={
        "index_id": "my-index",
        "vectors": vectors
    }
)

# Search
results = requests.post(
    f"{base_url}/search",
    headers=headers,
    json={
        "index_id": "my-index",
        "query_vector": query,
        "k": 10
    }
).json()

import grpc
from mlgraph_pb2 import *
from mlgraph_pb2_grpc import *

# Connect to server
channel = grpc.insecure_channel(
    'localhost:50051'
)
stub = MLGraphDaemonStub(channel)

# Create index
request = CreateIndexRequest(
    config=IndexConfig(
        index_id="my-index",
        dimension=768,
        index_type="IVFFlat",
        num_centroids=100
    )
)
response = stub.CreateIndex(request)

# Stream search
def search_stream():
    for query in queries:
        yield SearchRequest(
            index_id="my-index",
            query_vector=query,
            k=10
        )

for result in stub.StreamSearch(
    search_stream()
):
    print(f"Found {len(result.results)}")

const axios = require('axios');

const client = axios.create({
  baseURL: 'http://localhost:8080/api/v1',
  headers: {
    'Authorization': `Bearer ${token}`
  }
});

// Create index
await client.post('/indices', {
  index_id: 'my-index',
  dimension: 768,
  index_type: 'IVFFlat',
  num_centroids: 100
});

// Search
const { data } = await client.post('/search', {
  index_id: 'my-index',
  query_vector: queryVector,
  k: 10,
  include_metadata: true
});

console.log(`Found ${data.results.length}`);

# Login
curl -X POST http://localhost:8080/api/v1/auth/login \
  -H "Content-Type: application/json" \
  -d '{"email":"user@example.com","password":"pass"}'

# Create index
curl -X POST http://localhost:8080/api/v1/indices \
  -H "Authorization: Bearer $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"index_id":"my-index","dimension":768}'

# Search
curl -X POST http://localhost:8080/api/v1/search \
  -H "Authorization: Bearer $TOKEN" \
  -H "Content-Type: application/json" \
  -d '{"index_id":"my-index","query_vector":[...],"k":10}'