32 lines
672 B
Go
32 lines
672 B
Go
|
package llm
|
||
|
|
|
||
|
|
// Memory is a mock implementation of EmbedderCompleter
|
||
|
|
type Memory struct {
|
||
|
|
EmbedReturns [][]float32
|
||
|
|
EmbedError error
|
||
|
|
CompleteReturns []string
|
||
|
|
CompleteError error
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Memory) Embed(input string) ([]float32, error) {
|
||
|
|
if m.EmbedError != nil {
|
||
|
|
return nil, m.EmbedError
|
||
|
|
}
|
||
|
|
res := m.EmbedReturns[0]
|
||
|
|
if len(m.EmbedReturns) > 1 {
|
||
|
|
m.EmbedReturns = m.EmbedReturns[1:]
|
||
|
|
}
|
||
|
|
return res, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (m *Memory) Complete(input string) (string, error) {
|
||
|
|
if m.CompleteError != nil {
|
||
|
|
return "", m.CompleteError
|
||
|
|
}
|
||
|
|
res := m.CompleteReturns[0]
|
||
|
|
if len(m.CompleteReturns) > 1 {
|
||
|
|
m.CompleteReturns = m.CompleteReturns[1:]
|
||
|
|
}
|
||
|
|
return res, nil
|
||
|
|
}
|