diff --git a/backend/agents/client/client.go b/backend/agents/client/client.go index d2542c9..c4838df 100644 --- a/backend/agents/client/client.go +++ b/backend/agents/client/client.go @@ -246,7 +246,7 @@ func (client *AgentClient) RunAgent(userId uuid.UUID, imageId uuid.UUID, imageNa request := AgentRequestBody{ Tools: &tools, ToolChoice: &toolChoice, - Model: "google/gemini-2.5-flash", + Model: "policy/images", RandomSeed: &seed, Temperature: 0.3, EndToolCall: client.Options.EndToolCall, @@ -284,7 +284,7 @@ func (client *AgentClient) RunAgentAlone(userID uuid.UUID, userReq string) error request := AgentRequestBody{ Tools: &tools, ToolChoice: &toolChoice, - Model: "google/gemini-2.5-flash", + Model: "policy/images", RandomSeed: &seed, Temperature: 0.3, EndToolCall: client.Options.EndToolCall, diff --git a/backend/agents/create_list_agent.go b/backend/agents/create_list_agent.go index 75901f4..e1f7447 100644 --- a/backend/agents/create_list_agent.go +++ b/backend/agents/create_list_agent.go @@ -81,7 +81,7 @@ type CreateListAgent struct { func (agent *CreateListAgent) CreateList(log *log.Logger, userID uuid.UUID, userReq string) error { request := client.AgentRequestBody{ - Model: "google/gemini-2.5-flash", + Model: "policy/images", Temperature: 0.3, ResponseFormat: client.ResponseFormat{ Type: "json_object", diff --git a/backend/agents/description_agent.go b/backend/agents/description_agent.go index 169f645..bab10fd 100644 --- a/backend/agents/description_agent.go +++ b/backend/agents/description_agent.go @@ -28,7 +28,7 @@ type DescriptionAgent struct { func (agent DescriptionAgent) Describe(log *log.Logger, imageId uuid.UUID, imageName string, imageData []byte) error { request := client.AgentRequestBody{ - Model: "google/gemini-2.5-flash-lite-preview-06-17", + Model: "policy/images", Temperature: 0.3, ResponseFormat: client.ResponseFormat{ Type: "text",