ดาวน์โหลด memista - ดาวน์โหลดซอร์สโค้ด memista

memista

ซอร์สโค้ดอื่น ๆ

1.0.0

ดาวน์โหลด

MEMISTA: บริการค้นหาเวกเตอร์ประสิทธิภาพสูง

MEMISTA เป็นบริการค้นหาเวกเตอร์ประสิทธิภาพสูงที่เขียนในสนิมที่ให้ HTTP API ง่าย ๆ สำหรับการจัดเก็บและดึงข้อความข้อความด้วยการฝังเวกเตอร์ที่เกี่ยวข้อง มันรวม SQLite สำหรับการจัดเก็บข้อมูลเมตากับ USEARCH สำหรับการค้นหาความคล้ายคลึงกันของเวกเตอร์ที่มีประสิทธิภาพ

คุณสมบัติ

การค้นหาความคล้ายคลึงกันของเวกเตอร์อย่างรวดเร็วโดยใช้ USEARCH
การจัดเก็บชิ้นข้อความและข้อมูลเมตาใน sqlite อย่างต่อเนื่อง
การรองรับฐานข้อมูลหลายรายการผ่านการแบ่งพาร์ติชัน database_id
เอกสาร OpenAPI ที่มีตัวเลือก UI หลายตัว (Swagger, Redoc, Rapidoc)
กำหนดค่าได้ผ่านตัวแปรสภาพแวดล้อม
async I/O เพื่อประสิทธิภาพสูง

จุดสิ้นสุด API

โพสต์ /v1 /แทรก

แทรกชิ้นข้อความด้วยการฝังตัวลงในฐานข้อมูลที่ระบุ

โพสต์ /v1 /ค้นหา

ค้นหาชิ้นที่คล้ายกันโดยใช้เวกเตอร์ฝังตัว

ลบ /v1 /drop

วางฐานข้อมูลเฉพาะและดัชนีเวกเตอร์ที่เกี่ยวข้อง

การกำหนดค่า

บริการสามารถกำหนดค่าได้โดยใช้ตัวแปรสภาพแวดล้อม:

DATABASE_PATH : path ไปยังไฟล์ฐานข้อมูล sqlite (ค่าเริ่มต้น: "memista.db"))
SERVER_HOST : ที่อยู่โฮสต์ที่จะผูกกับ (ค่าเริ่มต้น: "127.0.0.1")
SERVER_PORT : พอร์ตเพื่อฟัง (ค่าเริ่มต้น: 8083)
LOG_LEVEL : ระดับการบันทึก (ค่าเริ่มต้น: "ข้อมูล")

เริ่มต้นอย่างรวดเร็ว

ติดตั้งสนิมและสินค้า
โคลนที่เก็บนี้
สร้างไฟล์ .env ด้วยการกำหนดค่าของคุณ (ไม่บังคับ)
เรียกใช้เซิร์ฟเวอร์:

cargo run

เซิร์ฟเวอร์จะเริ่มและเอกสาร API จะพร้อมใช้งานที่:

Swagger UI: http: // localhost: 8083/swagger
redoc: http: // localhost: 8083/redoc
Rapidoc: http: // localhost: 8083/rapidoc
openapi json: http: // localhost: 8083/openapi.json

ตัวอย่างการใช้งาน

แทรกชิ้น

curl -X POST http://localhost:8083/v1/insert 
  -H " Content-Type: application/json " 
  -d ' {
    "database_id": "my_db",
    "chunks": [{
      "embedding": [0.1, 0.2],
      "text": "Sample text",
      "metadata": "{"source": "document1"}"
    }]
  } '

ค้นหาชิ้น

curl -X POST http://localhost:8083/v1/search 
  -H " Content-Type: application/json " 
  -d ' {
    "database_id": "my_db",
    "embeddings": [[0.1, 0.2]],
    "num_results": 5
  } '