Added package

Author: ankane

.github/workflows/build.yml

@@ -21,5 +21,4 @@ jobs:
           make
           sudo make install
       - run: nimble -y install db_connector
-      - run: nim c example.nim
-      - run: ./example
+      - run: nimble test --features:dev --parser:declarative

CHANGELOG.md

@@ -0,0 +1,3 @@
+## 0.1.0 (unreleased)
+
+- First release

README.md

@@ -1,6 +1,6 @@
 # pgvector-nim
 
-[pgvector](https://github.com/pgvector/pgvector) examples for Nim
+[pgvector](https://github.com/pgvector/pgvector) support for Nim
 
 Supports [db_connector](https://github.com/nim-lang/db_connector)
 
@@ -63,7 +63,11 @@ db.exec(sql"CREATE INDEX ON items USING ivfflat (embedding vector_l2_ops) WITH (
 
 Use `vector_ip_ops` for inner product and `vector_cosine_ops` for cosine distance
 
-See a [full example](example.nim)
+See a [full example](tests/tdb_connector.nim)
+
+## History
+
+View the [changelog](https://github.com/pgvector/pgvector-nim/blob/master/CHANGELOG.md)
 
 ## Contributing
 
@@ -81,13 +85,13 @@ git clone https://github.com/pgvector/pgvector-nim.git
 cd pgvector-nim
 createdb pgvector_nim_test
 nimble install db_connector
-nim c --run example.nim
+nimble test --features:dev --parser:declarative
 ```
 
 Specify the path to libpq if needed:
 
 ```sh
-nim c --run --dynlibOverride:pq --passL:"/opt/homebrew/opt/libpq/lib/libpq.dylib" example.nim
+nimble test --features:dev --parser:declarative --dynlibOverride:pq --passL:"/opt/homebrew/opt/libpq/lib/libpq.dylib"
 ```
 
 To run an example:
@@ -97,3 +101,9 @@ cd examples/openai
 createdb pgvector_example
 nim c --run example.nim
 ```
+
+Specify the path to libpq if needed:
+
+```sh
+nim c --run --dynlibOverride:pq --passL:"/opt/homebrew/opt/libpq/lib/libpq.dylib" example.nim
+```

example.nim

@@ -0,0 +1,15 @@
+# Package
+
+version       = "0.1.0"
+author        = "Andrew Kane"
+description   = "pgvector support for Nim"
+license       = "MIT"
+srcDir        = "src"
+
+
+# Dependencies
+
+requires "nim >= 2.0.0"
+
+dev:
+  requires "db_connector >= 0.1.0"

pgvector.nimble

@@ -0,0 +1,3 @@
+import pgvector/[halfvec, sparsevec, vector]
+
+export halfvec, sparsevec, vector

src/pgvector.nim

@@ -0,0 +1,16 @@
+import std/[json, sequtils, sugar]
+
+type HalfVector* = object
+  vec: seq[float32]
+
+proc `$`*(v: HalfVector): string =
+  result = $(%* v.vec)
+
+proc toSeq*(v: HalfVector): seq[float32] =
+  result = v.vec
+
+proc toHalfVector*[T](a: openArray[T]): HalfVector =
+  result = HalfVector(vec: a.toSeq.map(v => float32(v)))
+
+proc toHalfVector*(s: string): HalfVector =
+  result = HalfVector(vec: to(parseJson(s), seq[float32]))

src/pgvector/halfvec.nim

@@ -0,0 +1,35 @@
+import std/[strutils, sugar, tables]
+
+type SparseVector* = object
+  elements*: Table[int, float32]
+  dim*: int
+
+proc `$`*(v: SparseVector): string =
+  let elements = collect(newSeqOfCap(v.elements.len)):
+    for k, v in v.elements: $(k + 1) & ":" & $v
+  result = "{" & elements.join(",") & "}/" & $v.dim
+
+proc toSparseVector*[T](a: openArray[T]): SparseVector =
+  var elements = initTable[int, float32]()
+  for i, v in a:
+    let f = float32(v)
+    if f != 0:
+      elements[i] = f
+  result = SparseVector(elements: elements, dim: a.len)
+
+proc toSparseVector*[T](elements: Table[int, T], dim: int): SparseVector =
+  var e = initTable[int, float32]()
+  for k, v in elements:
+    e[k] = float32(v)
+  result = SparseVector(elements: e, dim: dim)
+
+proc toSparseVector*(s: string): SparseVector =
+  let parts = s.split("}/", maxsplit = 2)
+  var elements = initTable[int, float32]()
+  for e in parts[0][1..^1].split(","):
+    let ep = e.split(":", maxsplit = 2)
+    let index = parseInt(ep[0])
+    let value = float32(parseFloat(ep[1]))
+    elements[index - 1] = value
+  let dim = parseInt(parts[1])
+  result = SparseVector(elements: elements, dim: dim)

src/pgvector/sparsevec.nim

@@ -0,0 +1,16 @@
+import std/[json, sequtils, sugar]
+
+type Vector* = object
+  vec: seq[float32]
+
+proc `$`*(v: Vector): string =
+  result = $(%* v.vec)
+
+proc toSeq*(v: Vector): seq[float32] =
+  result = v.vec
+
+proc toVector*[T](a: openArray[T]): Vector =
+  result = Vector(vec: a.toSeq.map(v => float32(v)))
+
+proc toVector*(s: string): Vector =
+  result = Vector(vec: to(parseJson(s), seq[float32]))

src/pgvector/vector.nim

@@ -0,0 +1 @@
+switch("path", "$projectDir/../src")