chandlerswift/svs-services-server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add testimonials

Browse source
This commit is contained in:
Chandler Swift 2025-12-28 19:42:39 -06:00
parent 06e943d478
commit 40223a12b5
7 changed files with 318 additions and 101 deletions
Download patch file Download diff file Expand all files Collapse all files

27
completion/openrouter/openrouter.go
View file

@ -22,16 +22,13 @@ type Message struct {
}
type ChatCompletionRequest struct {
Model string `json:"model"`
Messages []Message `json:"messages"`
Temperature *float64 `json:"temperature,omitempty"`
MaxTokens *int `json:"max_tokens,omitempty"`
TopP *float64 `json:"top_p,omitempty"`
Stop json.RawMessage `json:"stop,omitempty"` // string or []string; keep flexible
Provider struct {
Model string `json:"model"`
Messages []Message `json:"messages"`
Provider struct {
Sort string `json:"sort,omitempty"`
} `json:"provider,omitempty"`
ResponseFormat any `json:"response_format,omitempty"` // keep flexible
Plugins []map[string]string `json:"plugins,omitempty"`
}
type ChatCompletionResponse struct {
@ -46,7 +43,7 @@ type ChatCompletionResponse struct {
} `json:"choices"`
}
func (p OpenRouterProvider) Complete(ctx context.Context, prompt string) (string, error) {
func (p OpenRouterProvider) Complete(ctx context.Context, prompt string, schema any) (string, error) {
req := ChatCompletionRequest{
Model: p.Model,
Messages: []Message{
@ -62,6 +59,18 @@ func (p OpenRouterProvider) Complete(ctx context.Context, prompt string) (string
},
}
if schema != nil {
req.ResponseFormat = map[string]any{
"type": "json_schema",
"json_schema": schema,
}
req.Plugins = []map[string]string{
{
"id": "response-healing",
},
}
}
httpClient := http.Client{Timeout: 10 * time.Second}
body, err := json.Marshal(req)
if err != nil {