From 78e3be4575df71a92f0694aa78d256557e75d6a9 Mon Sep 17 00:00:00 2001
From: Sakurasan <26715255+Sakurasan@users.noreply.github.com>
Date: Sun, 17 Nov 2024 22:32:14 +0800
Subject: [PATCH] update: allow cors

---
 go.mod               | 1 +
 go.sum               | 2 ++
 opencat.go           | 3 ++-
 pkg/vertexai/auth.go | 3 +++
 4 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/go.mod b/go.mod
index 30c21df..26b94b9 100644
--- a/go.mod
+++ b/go.mod
@@ -42,6 +42,7 @@ require (
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/felixge/httpsnoop v1.0.4 // indirect
 	github.com/gabriel-vasile/mimetype v1.4.6 // indirect
+	github.com/gin-contrib/cors v1.7.2 // indirect
 	github.com/gin-contrib/sse v0.1.0 // indirect
 	github.com/glebarez/go-sqlite v1.22.0 // indirect
 	github.com/go-logr/logr v1.4.2 // indirect
diff --git a/go.sum b/go.sum
index 89930f8..3548eb6 100644
--- a/go.sum
+++ b/go.sum
@@ -69,6 +69,8 @@ github.com/gabriel-vasile/mimetype v1.4.6 h1:3+PzJTKLkvgjeTbts6msPJt4DixhT4YtFNf
 github.com/gabriel-vasile/mimetype v1.4.6/go.mod h1:JX1qVKqZd40hUPpAfiNTe0Sne7hdfKSbOqqmkq8GCXc=
 github.com/gdamore/encoding v1.0.0/go.mod h1:alR0ol34c49FCSBLjhosxzcPHQbf2trDkoo5dl+VrEg=
 github.com/gdamore/tcell v1.3.0/go.mod h1:Hjvr+Ofd+gLglo7RYKxxnzCBmev3BzsS67MebKS4zMM=
+github.com/gin-contrib/cors v1.7.2 h1:oLDHxdg8W/XDoN/8zamqk/Drgt4oVZDvaV0YmvVICQw=
+github.com/gin-contrib/cors v1.7.2/go.mod h1:SUJVARKgQ40dmrzgXEVxj2m7Ig1v1qIboQkPDTQ9t2E=
 github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
 github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
 github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
diff --git a/opencat.go b/opencat.go
index a3ebfe5..fecc219 100644
--- a/opencat.go
+++ b/opencat.go
@@ -14,6 +14,7 @@ import (
 	"os"
 
 	"github.com/duke-git/lancet/v2/fileutil"
+	"github.com/gin-contrib/cors"
 	"github.com/gin-gonic/gin"
 	"github.com/google/uuid"
 	"gorm.io/gorm"
@@ -166,7 +167,7 @@ func main() {
 	// 初始化用户
 	r.POST("/1/users/init", team.Handleinit)
 
-	r.Any("/v1/*proxypath", router.HandleProxy)
+	r.Any("/v1/*proxypath", cors.Default(), router.HandleProxy)
 
 	// r.POST("/v1/chat/completions", router.HandleProy)
 	// r.GET("/v1/models", router.HandleProy)
diff --git a/pkg/vertexai/auth.go b/pkg/vertexai/auth.go
index 7edb293..28a8d08 100644
--- a/pkg/vertexai/auth.go
+++ b/pkg/vertexai/auth.go
@@ -19,6 +19,9 @@ https://$LOCATION-aiplatform.googleapis.com/v1/projects/${PROJECT_ID}/locations/
   "stream": true,
   "max_tokens": 4096
 }'
+
+quota:
+https://console.cloud.google.com/iam-admin/quotas?hl=zh-cn
 */
 
 package vertexai