JavaScript API PS

README.md

@@ -187,6 +187,14 @@ ollama.embeddings(request)
   - `options` `<Options>`: (Optional) Options to configure the runtime.
 - Returns: `<EmbeddingsResponse>`
 
+### ps
+
+```javascript
+ollama.ps()
+```
+
+- Returns: `<ListResponse>`
+
 ## Custom client
 
 A custom client can be created with the following fields:

src/browser.ts

@@ -63,7 +63,7 @@
    */
   protected async processStreamableRequest<T extends object>(
     endpoint: string,
     request: { stream?: boolean } & Record<string, any>,
   ): Promise<T | AsyncGenerator<T>> {
     request.stream = request.stream ?? false
     const response = await utils.post(
@@ -90,7 +90,7 @@
           yield message
           // message will be done in the case of chat and generate
           // message will be success in the case of a progress response (pull, push, create)
           if ((message as any).done || (message as any).status === 'success') {
             return
           }
         }
@@ -98,7 +98,7 @@
       })()
     } else {
       const message = await itr.next()
       if (!message.value.done && (message.value as any).status !== 'success') {
         throw new Error('Expected a completed response.')
       }
       return message.value
@@ -278,6 +278,16 @@
     })
     return (await response.json()) as EmbeddingsResponse
   }
+
+  /**
+   * Lists the running models on the server
+   * @returns {Promise<ListResponse>} - The response object.
+   * @throws {Error} - If the response body is missing.
+   */
+  async ps(): Promise<ListResponse> {
+    const response = await utils.get(this.fetch, `${this.config.host}/api/ps`)
+    return (await response.json()) as ListResponse
+  }
 }
 
 export default new Ollama()

src/interfaces.ts

@@ -166,6 +166,8 @@ export interface ModelResponse {
   size: number
   digest: string
   details: ModelDetails
+  expires_at: Date
+  size_vram: number
 }
 
 export interface ModelDetails {