mirror of
https://github.com/mudler/cogito.git
synced 2026-07-24 19:05:32 -04:00
2f20fcce01
* Split e2e and unit tests Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Abstract away LLM to an interface Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Update README Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Fixups Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Enhance content improve generation Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Add unit test for reviewer Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Better handling of result status of tool calls Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Fixups Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * Add tools test file Signed-off-by: Ettore Di Giacinto <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
13 lines
276 B
Go
13 lines
276 B
Go
package cogito
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/sashabaranov/go-openai"
|
|
)
|
|
|
|
type LLM interface {
|
|
Ask(ctx context.Context, f Fragment) (Fragment, error)
|
|
CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error)
|
|
}
|