mirror of https://github.com/chaitin/PandaWiki.git
34 lines
1.0 KiB
Go
34 lines
1.0 KiB
Go
package rag
|
||
|
||
import (
|
||
"context"
|
||
)
|
||
|
||
// RetrieveChunks 检索分块（向量/关键词检索）
|
||
func (c *Client) RetrieveChunks(ctx context.Context, req RetrievalRequest) ([]RetrievalChunk, int, string, error) {
|
||
httpReq, err := c.newRequest(ctx, "POST", "retrieval", req)
|
||
if err != nil {
|
||
return nil, 0, "", err
|
||
}
|
||
var resp RetrievalResponse
|
||
if err := c.do(httpReq, &resp); err != nil {
|
||
return nil, 0, "", err
|
||
}
|
||
return resp.Data.Chunks, resp.Data.Total, resp.Data.RewrittenQuery, nil
|
||
}
|
||
|
||
// RelatedQuestions 生成相关问题（多样化检索）
|
||
// 注意：该接口需要 Bearer Login Token，通常与API Key不同
|
||
func (c *Client) RelatedQuestions(ctx context.Context, loginToken string, req RelatedQuestionsRequest) ([]string, error) {
|
||
httpReq, err := c.newRequest(ctx, "POST", "/v1/conversation/related_questions", req)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
httpReq.Header.Set("Authorization", "Bearer "+loginToken)
|
||
var resp RelatedQuestionsResponse
|
||
if err := c.do(httpReq, &resp); err != nil {
|
||
return nil, err
|
||
}
|
||
return resp.Data, nil
|
||
}
|