H-337: restore Captain inbox takeover and KB citations (#65)
* fix(captain): restore inbox takeover and KB citations * fix(captain): harden grounded citations and smoke seed --------- Co-authored-by: Rogee <rogee@ipao.vip>
This commit is contained in:
@@ -214,6 +214,7 @@ func setupCaptainAssistantHandlerTestWithProvider(t *testing.T, provider llm.Pro
|
||||
&model.Inbox{},
|
||||
&model.CaptainAssistant{},
|
||||
&model.CaptainInbox{},
|
||||
&model.CaptainDocument{},
|
||||
&model.CaptainAssistantResponse{},
|
||||
))
|
||||
t.Cleanup(func() {
|
||||
@@ -534,7 +535,7 @@ func (p *captainPlaygroundFakeProvider) CreateEmbedding(ctx context.Context, req
|
||||
if p.embeddingErr != nil || p.embeddingResponseSet || p.embeddingResponse != nil {
|
||||
return p.embeddingResponse, p.embeddingErr
|
||||
}
|
||||
return &llm.EmbeddingResponse{}, nil
|
||||
return &llm.EmbeddingResponse{Data: []llm.EmbeddingData{{Embedding: []float64{0.1, 0.2, 0.3}}}}, nil
|
||||
}
|
||||
|
||||
func (p *captainPlaygroundFakeProvider) ChatCompletionStream(ctx context.Context, req llm.ChatRequest, onChunk func(llm.StreamChunk) error) error {
|
||||
|
||||
Reference in New Issue
Block a user