feat: new package model

packages/model/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "@xsai/model",
+  "version": "0.0.15",
+  "type": "module",
+  "author": "Moeru AI",
+  "license": "MIT",
+  "homepage": "https://xsai.js.org",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/moeru-ai/xsai.git",
+    "directory": "packages/model"
+  },
+  "bugs": "https://github.com/moeru-ai/xsai/issues",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "default": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "pkgroll",
+    "build:watch": "pkgroll --watch",
+    "test": "vitest run",
+    "test:watch": "vitest"
+  },
+  "dependencies": {
+    "@xsai/shared": "workspace:"
+  },
+  "devDependencies": {
+    "@xsai/providers": "workspace:"
+  }
+}

packages/model/src/index.ts

@@ -0,0 +1,2 @@
+export type * from './types'
+export * from './utils'

packages/model/src/types/index.ts

@@ -0,0 +1 @@
+export type { Model } from './model'

packages/model/src/types/model.ts

@@ -0,0 +1,6 @@
+export interface Model {
+  created: number
+  id: string
+  object: 'model'
+  owned_by: string
+}

packages/model/src/utils/index.ts

@@ -0,0 +1,2 @@
+export * from './list-models'
+export * from './retrieve-model'

packages/model/src/utils/list-models.ts

@@ -0,0 +1,21 @@
+import { type CommonRequestOptions, requestHeaders } from '@xsai/shared'
+
+import type { Model } from '../types/model'
+
+export interface ListModelsOptions extends Omit<CommonRequestOptions, 'model'> {}
+
+export interface ListModelsResponse {
+  data: Model[]
+  object: 'list'
+}
+
+export const listModels = async (options: ListModelsOptions): Promise<Model[]> =>
+  await fetch(options.url, {
+    headers: requestHeaders({
+      'Content-Type': 'application/json',
+      ...options.headers,
+    }, options.apiKey),
+    signal: options.abortSignal,
+  })
+    .then(res => res.json() as Promise<ListModelsResponse>)
+    .then(({ data }) => data)

packages/model/src/utils/retrieve-model.ts

@@ -0,0 +1,15 @@
+import { type CommonRequestOptions, requestHeaders } from '@xsai/shared'
+
+import type { Model } from '../types/model'
+
+export interface RetrieveModelOptions extends Omit<CommonRequestOptions, 'model'> {}
+
+export const retrieveModel = async (options: RetrieveModelOptions): Promise<Model> =>
+  await fetch(options.url, {
+    headers: requestHeaders({
+      'Content-Type': 'application/json',
+      ...options.headers,
+    }, options.apiKey),
+    signal: options.abortSignal,
+  })
+    .then(res => res.json() as Promise<Model>)

packages/model/test/index.test.ts

@@ -0,0 +1,27 @@
+import { describe, expect, it } from 'vitest'
+
+import { listModels, retrieveModel } from '../src'
+
+describe('@xsai/model', () => {
+  it('listModels', async () => {
+    const models = await listModels({
+      url: new URL('models', 'http://localhost:11434/v1/'),
+    })
+
+    const llama = models.find(({ id }) => id === 'llama3.2:latest')!
+
+    expect(llama.id).toBe('llama3.2:latest')
+    expect(llama.object).toBe('model')
+    expect(llama.owned_by).toBe('library')
+  })
+
+  it('retrieveModel', async () => {
+    const llama = await retrieveModel({
+      url: new URL('models/llama3.2:latest', 'http://localhost:11434/v1/'),
+    })
+
+    expect(llama.id).toBe('llama3.2:latest')
+    expect(llama.object).toBe('model')
+    expect(llama.owned_by).toBe('library')
+  })
+})

packages/model/tsconfig.json

@@ -0,0 +1,4 @@
+{
+  "extends": "@importantimport/tsconfig/app.json",
+  "include": ["src"]
+}