accept all 2xx http code, increase timeout
This commit is contained in:
+2
-2
@@ -20,7 +20,7 @@ type Client struct {
|
|||||||
|
|
||||||
func NewClient(env *env.Env) *Client {
|
func NewClient(env *env.Env) *Client {
|
||||||
return &Client{
|
return &Client{
|
||||||
http: http.Client{Timeout: 10 * time.Second},
|
http: http.Client{Timeout: 30 * time.Second},
|
||||||
token: env.LlmToken,
|
token: env.LlmToken,
|
||||||
model: env.LlmModel,
|
model: env.LlmModel,
|
||||||
endpoint: env.LlmEndpoint,
|
endpoint: env.LlmEndpoint,
|
||||||
@@ -52,7 +52,7 @@ func (client Client) GenerateContent(message string) (string, error) {
|
|||||||
|
|
||||||
body, err := io.ReadAll(res.Body)
|
body, err := io.ReadAll(res.Body)
|
||||||
|
|
||||||
if res.StatusCode != 200 {
|
if res.StatusCode >= 400 {
|
||||||
return "", fmt.Errorf("LLM returned status code [%d]", res.StatusCode)
|
return "", fmt.Errorf("LLM returned status code [%d]", res.StatusCode)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user