pi-media-models 0.1.7 → 0.1.8

package/README.md

@@ -101,7 +101,7 @@ Create or edit `~/.pi/agent/media-models.json`:
 }
 ```
 
-> **Note**: You only need to fill in the providers you plan to use. Unused providers can simply be omitted. Vertex reads `project_id` from the service-account JSON when `project` is omitted or still set to a placeholder such as `my-gcp-project`; an explicit real project ID remains supported.
+> **Note**: You only need to fill in the providers you plan to use. Unused providers can simply be omitted. Vertex reads `project_id` from the service-account JSON when `project` is omitted or still set to a placeholder such as `my-gcp-project`; an explicit real project ID remains supported. Set `location` to `global` to use Vertex's global endpoint for models that support it.
 
 ### Custom OpenAI-Compatible Providers
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-media-models",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "type": "module",
   "description": "Provider-neutral multimodal generation tools for Pi Coding Agent",
   "keywords": [

package/src/adapters/google.ts

@@ -212,7 +212,7 @@ export class GoogleMediaAdapter extends BaseAdapter {
     const models: ModelDescriptor[] = []
     let pageToken: string | undefined
     do {
-      const url = new URL(`https://${location}-aiplatform.googleapis.com/v1beta1/publishers/google/models`)
+      const url = new URL(`${vertexApiOrigin(location)}/v1beta1/publishers/google/models`)
       url.searchParams.set('pageSize', '100')
       url.searchParams.set('listAllVersions', 'true')
       if (pageToken) url.searchParams.set('pageToken', pageToken)
@@ -247,7 +247,7 @@ export class GoogleMediaAdapter extends BaseAdapter {
       return { url: `${base}/models/${encodeURIComponent(model)}:${method}`, headers: { 'x-goog-api-key': key }, operationsBase: base }
     }
     const { auth, project, location } = await this.vertexSettings(options)
-    const base = `https://${location}-aiplatform.googleapis.com/v1`
+    const base = `${vertexApiOrigin(location)}/v1`
     const resource = `projects/${encodeURIComponent(project)}/locations/${encodeURIComponent(location)}/publishers/google/models/${encodeURIComponent(model)}`
     const url = `${base}/${resource}:${method}`
     const authHeaders = await auth.getRequestHeaders(url)
@@ -286,6 +286,10 @@ export class GoogleMediaAdapter extends BaseAdapter {
   }
 }
 
+export function vertexApiOrigin(location: string): string {
+  return location === 'global' ? 'https://aiplatform.googleapis.com' : `https://${location}-aiplatform.googleapis.com`
+}
+
 function googleCapabilities(id: string, declared: ModelDescriptor[]): Capability[] {
   const known = declared.find(model => model.id === id)
   if (known) return known.capabilities