From 2279916a4463cc29a69678a0793bd03ddd7d68dc Mon Sep 17 00:00:00 2001
From: renzhiyuan <465386466@qq.com>
Date: Tue, 4 Mar 2025 14:21:37 +0800
Subject: [PATCH] first commit

---
 ai.go                 | 32 +++++++++++++++++++++++++
 ai/doubao/constant.go | 17 ++++++++++++++
 ai/doubao/doubao.go   | 54 +++++++++++++++++++++++++++++++++++++++++++
 ai_test.go            | 11 +++++++++
 go.mod                | 12 ++++++++++
 5 files changed, 126 insertions(+)
 create mode 100644 ai.go
 create mode 100644 ai/doubao/constant.go
 create mode 100644 ai/doubao/doubao.go
 create mode 100644 ai_test.go
 create mode 100644 go.mod

diff --git a/ai.go b/ai.go
new file mode 100644
index 0000000..dfa1310
--- /dev/null
+++ b/ai.go
@@ -0,0 +1,32 @@
+package l_ai_address
+
+import (
+	"context"
+	"gitea.cdlsxd.cn/self-tools/l_ai_address/ai/doubao"
+	"strings"
+)
+
+type Address struct {
+	P string
+	C string
+	D string
+}
+
+func GetAddress(ctx context.Context, address, key, model string) (Address, error) {
+
+	modelObj := doubao.NewDouBao(model, key)
+	text := []string{
+		"解析省市区地址[QUESTION]" + address + "[/QUESTION]",
+		"-根据国家统计局的地址信息 帮我匹配出对应的省市区信息，如果是直辖县级行政区，直接按照区来处理，然后市一级显示上级代管市",
+		"-只需要给出省市区具体名称，不需要解释",
+		"-省市区之间用\",\"隔开",
+	}
+	res, err := modelObj.GetData(ctx, doubao.Text, func(input string) (string, error) {
+		return input, nil
+	}, text...)
+	if err != nil {
+		return Address{}, err
+	}
+	resSlice := strings.Split(res, ",")
+	return Address{P: resSlice[0], C: resSlice[1], D: resSlice[2]}, nil
+}
diff --git a/ai/doubao/constant.go b/ai/doubao/constant.go
new file mode 100644
index 0000000..1f00ea6
--- /dev/null
+++ b/ai/doubao/constant.go
@@ -0,0 +1,17 @@
+package doubao
+
+var UrlMap = map[UrlType]string{
+	Text:      "https://ark.cn-beijing.volces.com/api/v3/chat/completions",
+	Video:     "https://ark.cn-beijing.volces.com/api/v3/contents/generations/tasks",
+	Embedding: "https://ark.cn-beijing.volces.com/api/v3/embeddings",
+	Token:     "https://ark.cn-beijing.volces.com/api/v3/tokenization",
+}
+
+type UrlType string
+
+const (
+	Text      UrlType = "text"
+	Video     UrlType = "video"
+	Embedding UrlType = "embedding"
+	Token     UrlType = "token"
+)
diff --git a/ai/doubao/doubao.go b/ai/doubao/doubao.go
new file mode 100644
index 0000000..32399bf
--- /dev/null
+++ b/ai/doubao/doubao.go
@@ -0,0 +1,54 @@
+package doubao
+
+import (
+	"context"
+	"github.com/volcengine/volcengine-go-sdk/service/arkruntime"
+	"github.com/volcengine/volcengine-go-sdk/service/arkruntime/model"
+	"github.com/volcengine/volcengine-go-sdk/volcengine"
+	"time"
+)
+
+type DouBao struct {
+	Model string
+	Key   string
+}
+
+func NewDouBao(modelName string, key string) *DouBao {
+	return &DouBao{
+		Model: modelName,
+		Key:   key,
+	}
+}
+
+func (o *DouBao) GetData(ctx context.Context, url UrlType, respHandle func(input string) (string, error), text ...string) (string, error) {
+	var Message = make([]*model.ChatCompletionMessage, len(text))
+
+	key := "914ccf1d-c002-4fad-a431-f291f5e0d2ad"
+	client := arkruntime.NewClientWithApiKey(
+		key,
+		//arkruntime.WithBaseUrl(UrlMap[url]),
+		arkruntime.WithRegion("cn-beijing"),
+		arkruntime.WithTimeout(2*time.Minute),
+		arkruntime.WithRetryTimes(2),
+	)
+	for k, v := range text {
+		Message[k] = &model.ChatCompletionMessage{
+			Role: model.ChatMessageRoleSystem,
+			Content: &model.ChatCompletionMessageContent{
+				StringValue: volcengine.String(v),
+			},
+		}
+	}
+	req := model.CreateChatCompletionRequest{
+		Model:    o.Model,
+		Messages: Message,
+	}
+
+	resp, err := client.CreateChatCompletion(ctx, req)
+	if err != nil {
+		return "", err
+	}
+	result, err := respHandle(*resp.Choices[0].Message.Content.StringValue)
+
+	return result, err
+}
diff --git a/ai_test.go b/ai_test.go
new file mode 100644
index 0000000..fa4e8eb
--- /dev/null
+++ b/ai_test.go
@@ -0,0 +1,11 @@
+package l_ai_address
+
+import (
+	"context"
+	"testing"
+)
+
+func TestAddress(t *testing.T) {
+	res, err := GetAddress(context.Background(), "江苏省南通市海门市万科翡翠甲第15幢1101室", "236ba4b6-9daa-4755-b22f-2fd274cd223a", "doubao-1-5-lite-32k-250115")
+	t.Log(res, err)
+}
diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..237c83d
--- /dev/null
+++ b/go.mod
@@ -0,0 +1,12 @@
+module gitea.cdlsxd.cn/self-tools/l_ai_address
+
+go 1.23.6
+
+require github.com/volcengine/volcengine-go-sdk v1.0.184
+
+require (
+	github.com/google/uuid v1.3.0 // indirect
+	github.com/jmespath/go-jmespath v0.4.0 // indirect
+	github.com/volcengine/volc-sdk-golang v1.0.23 // indirect
+	gopkg.in/yaml.v2 v2.2.8 // indirect
+)