diff --git a/README.md b/README.md index e923f1f..b609ad0 100644 --- a/README.md +++ b/README.md @@ -3,13 +3,24 @@ Lightweight OpenAI/OLLaMA/GenAI compatible caller, originally designed for DeepI ## Quick Start ```go -api := NewAPI(OpenAIDeepInfra, "PUT YOUR API KEY HERE") -txt, err := api.Request(model.NewOpenAI(model.ModelDeepDeek, model.SeparatorThink, 0.7, 0.9, 1024). - System("Be a good assistant.").User("Hello"), +package main + +import ( + "fmt" + + "github.com/fumiama/deepinfra" + "github.com/fumiama/deepinfra/model" ) -if err != nil { - panic(err) + +func main() { + api := deepinfra.NewAPI(deepinfra.OpenAIDeepInfra, "PUT YOUR API KEY HERE") + txt, err := api.Request(model.NewOpenAI(model.ModelDeepDeek, model.SeparatorThink, 0.7, 0.9, 1024). + System("Be a good assistant.").User(model.NewContentText("Hello")), + ) + if err != nil { + panic(err) + } + fmt.Println(txt) + // Hello! How can I assist you today? } -fmt.Println(txt) -// Hello! How can I assist you today? ```