Merge pull request #632 from Abirdcfly/bing

feat: add bingsearch tools
kubeagi · Jan 25, 2024 · 55ae7da · 55ae7da
2 parents c537383 + 22ba5c4
commit 55ae7da
Show file tree

Hide file tree

Showing 31 changed files with 1,172 additions and 21 deletions.
diff --git a/api/app-node/chain/v1alpha1/llmchain_types.go b/api/app-node/chain/v1alpha1/llmchain_types.go
@@ -20,6 +20,7 @@ import (
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
 
 	node "github.com/kubeagi/arcadia/api/app-node"
+	agent "github.com/kubeagi/arcadia/api/app-node/agent/v1alpha1"
 	"github.com/kubeagi/arcadia/api/base/v1alpha1"
 )
 
@@ -31,6 +32,7 @@ type LLMChainSpec struct {
 }
 
 type CommonChainConfig struct {
+	Tools []agent.Tool `json:"tools,omitempty"`
 	// for memory
 	Memory Memory `json:"memory,omitempty"`
 

diff --git a/api/app-node/chain/v1alpha1/zz_generated.deepcopy.go b/api/app-node/chain/v1alpha1/zz_generated.deepcopy.go
diff --git a/apiserver/docs/docs.go b/apiserver/docs/docs.go
@@ -1174,7 +1174,7 @@ const docTemplate = `{
                     "example": "旷工最小计算单位为 0.5 天。"
                 },
                 "content": {
-                    "description": "related content in the source file",
+                    "description": "related content in the source file or in webpage",
                     "type": "string",
                     "example": "旷工最小计算单位为0.5天，不足0.5天以0.5天计算，超过0.5天不满1天以1天计算，以此类推。"
                 },
@@ -1207,6 +1207,16 @@ const docTemplate = `{
                     "description": "vector search score",
                     "type": "number",
                     "example": 0.34
+                },
+                "title": {
+                    "description": "Title of the webpage",
+                    "type": "string",
+                    "example": "开始使用 Microsoft 帐户 – Microsoft"
+                },
+                "url": {
+                    "description": "URL of the webpage",
+                    "type": "string",
+                    "example": "https://www.microsoft.com/zh-cn/welcome"
                 }
             }
         },

diff --git a/apiserver/docs/swagger.json b/apiserver/docs/swagger.json
@@ -1163,7 +1163,7 @@
                     "example": "旷工最小计算单位为 0.5 天。"
                 },
                 "content": {
-                    "description": "related content in the source file",
+                    "description": "related content in the source file or in webpage",
                     "type": "string",
                     "example": "旷工最小计算单位为0.5天，不足0.5天以0.5天计算，超过0.5天不满1天以1天计算，以此类推。"
                 },
@@ -1196,6 +1196,16 @@
                     "description": "vector search score",
                     "type": "number",
                     "example": 0.34
+                },
+                "title": {
+                    "description": "Title of the webpage",
+                    "type": "string",
+                    "example": "开始使用 Microsoft 帐户 – Microsoft"
+                },
+                "url": {
+                    "description": "URL of the webpage",
+                    "type": "string",
+                    "example": "https://www.microsoft.com/zh-cn/welcome"
                 }
             }
         },

diff --git a/apiserver/docs/swagger.yaml b/apiserver/docs/swagger.yaml
@@ -183,7 +183,7 @@ definitions:
         example: 旷工最小计算单位为 0.5 天。
         type: string
       content:
-        description: related content in the source file
+        description: related content in the source file or in webpage
         example: 旷工最小计算单位为0.5天，不足0.5天以0.5天计算，超过0.5天不满1天以1天计算，以此类推。
         type: string
       file_name:
@@ -210,6 +210,14 @@ definitions:
         description: vector search score
         example: 0.34
         type: number
+      title:
+        description: Title of the webpage
+        example: 开始使用 Microsoft 帐户 – Microsoft
+        type: string
+      url:
+        description: URL of the webpage
+        example: https://www.microsoft.com/zh-cn/welcome
+        type: string
     type: object
   service.Chunk:
     properties: