diff --git a/ai.go b/ai.go index b4aef8f..57a9d76 100644 --- a/ai.go +++ b/ai.go @@ -3,8 +3,8 @@ package l_ai_excel_header_match import ( "context" "encoding/json" + "gitea.cdlsxd.cn/self-tools/l_ai_excel_header_match/doubao" - "gitea.cdlsxd.cn/self-tools/l_ai_address/ai/doubao" "strings" ) diff --git a/doubao/doubao.go b/doubao/doubao.go index 5083198..bd90ef7 100644 --- a/doubao/doubao.go +++ b/doubao/doubao.go @@ -20,7 +20,7 @@ func NewDouBao(modelName string, key string) *DouBao { } } -func (o *DouBao) GetData(ctx context.Context, respHandle func(input string) (string, error), text ...string) (string, error) { +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)) client := arkruntime.NewClientWithApiKey(