alihan/chatgpt-cli
1
0
Fork 0
mirror of https://github.com/kardolus/chatgpt-cli.git synced 2024-09-08 23:15:00 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add debugging to the list-models command

Browse Source
This commit is contained in:
kardolus
2024-08-17 12:51:14 -04:00
parent e0cafe4d4d
commit e789c84249
1 changed files with 22 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
client/client.go
View File

@@ -74,7 +74,17 @@ func (c *Client) WithServiceURL(url string) *Client {
func (c *Client) ListModels() ([]string, error) {
var result []string
endpoint := c.getEndpoint(c.Config.ModelsPath)
if c.Config.Debug {
c.printRequestDebugInfo(endpoint, nil)
}
raw, err := c.caller.Get(c.getEndpoint(c.Config.ModelsPath))
if c.Config.Debug {
c.printResponseDebugInfo(raw)
}
if err != nil {
return nil, err
}
@@ -335,13 +345,20 @@ func generateUniqueSlug() string {
}
func (c *Client) printRequestDebugInfo(endpoint string, body []byte) {
bodyString := strings.ReplaceAll(string(body), "'", "'\"'\"'") // Escape single quotes
fmt.Printf("\nGenerated cURL command:\n\n")
fmt.Printf("curl --location --insecure --request POST '%s' \\\n", endpoint)
method := "POST"
if body == nil {
method = "GET"
}
fmt.Printf("curl --location --insecure --request %s '%s' \\\n", method, endpoint)
fmt.Printf(" --header \"Authorization: Bearer ${%s_API_KEY}\" \\\n", strings.ToUpper(c.Config.Name))
fmt.Printf(" --header 'Content-Type: application/json' \\\n")
fmt.Printf(" --data-raw '%s'\n\n", bodyString)
fmt.Printf(" --header 'Content-Type: application/json'")
if body != nil {
bodyString := strings.ReplaceAll(string(body), "'", "'\"'\"'") // Escape single quotes
fmt.Printf(" \\\n --data-raw '%s'", bodyString)
}
fmt.Println() // Print a newline at the end
}
func (c *Client) printResponseDebugInfo(raw []byte) {