clear comments

This commit is contained in:
Roy Han 2024-07-03 14:05:34 -07:00
parent 17de2b4405
commit 6caac01494
2 changed files with 0 additions and 5 deletions

View File

@ -33,7 +33,6 @@ type LlamaServer interface {
Ping(ctx context.Context) error
WaitUntilRunning(ctx context.Context) error
Completion(ctx context.Context, req CompletionRequest, fn func(CompletionResponse)) error
// Embedding(ctx context.Context, prompt string) ([]float64, error)
Embed(ctx context.Context, input []string) ([][]float32, error)
Tokenize(ctx context.Context, content string) ([]int, error)
Detokenize(ctx context.Context, tokens []int) (string, error)

View File

@ -648,10 +648,6 @@ func (s *mockLlm) WaitUntilRunning(ctx context.Context) error { return s.waitRes
func (s *mockLlm) Completion(ctx context.Context, req llm.CompletionRequest, fn func(llm.CompletionResponse)) error {
return s.completionResp
}
// func (s *mockLlm) Embedding(ctx context.Context, prompt string) ([]float64, error) {
// return s.embeddingResp, s.embeddingRespErr
// }
func (s *mockLlm) Embed(ctx context.Context, input []string) ([][]float32, error) {
return s.embedResp, s.embedRespErr
}