25 lines
408 B
Go
25 lines
408 B
Go
package llm
|
|
|
|
type OpenRoute struct {
|
|
APIKey string
|
|
Endpoint string
|
|
Model string
|
|
}
|
|
|
|
func NewOpenRoute(apiKey string, endpoint string, model string) *OpenAI {
|
|
return &OpenAI{
|
|
APIKey: apiKey,
|
|
Endpoint: endpoint,
|
|
Model: model,
|
|
}
|
|
}
|
|
|
|
func (o *OpenRoute) Generate(prompt string) (string, error) {
|
|
return "", nil
|
|
// TODO: implement
|
|
}
|
|
|
|
func (o *OpenRoute) GetModel() string {
|
|
return o.Model
|
|
}
|