diff --git a/src/components/agents/CreateAgentDialog.tsx b/src/components/agents/CreateAgentDialog.tsx
index ee6adc6..56a2858 100644
--- a/src/components/agents/CreateAgentDialog.tsx
+++ b/src/components/agents/CreateAgentDialog.tsx
@@ -1,6 +1,6 @@
import { useState } from "react";
import { useNavigate } from "react-router-dom";
-import { useMutation, useQueryClient } from "@tanstack/react-query";
+import { useMutation, useQuery, useQueryClient } from "@tanstack/react-query";
import {
Dialog,
DialogContent,
@@ -19,8 +19,7 @@ import {
SelectValue,
} from "@/components/ui/select";
import { useToast } from "@/hooks/use-toast";
-import { createAgent } from "@/lib/api/agents";
-import { AI_MODELS, getModelIdByValue } from "@/lib/constants/aiModels";
+import { createAgent, getAIModels } from "@/lib/api/agents";
interface CreateAgentDialogProps {
open: boolean;
@@ -32,7 +31,13 @@ export function CreateAgentDialog({ open, onOpenChange }: CreateAgentDialogProps
const { toast } = useToast();
const queryClient = useQueryClient();
const [name, setName] = useState("");
- const [selectedModel, setSelectedModel] = useState("");
+ const [selectedModelId, setSelectedModelId] = useState("");
+
+ // Buscar modelos da API
+ const { data: aiModels = [], isLoading: isLoadingModels } = useQuery({
+ queryKey: ["aiModels"],
+ queryFn: getAIModels,
+ });
const createAgentMutation = useMutation({
mutationFn: ({ name, modelId }: { name: string; modelId: number }) =>
@@ -53,7 +58,7 @@ export function CreateAgentDialog({ open, onOpenChange }: CreateAgentDialogProps
// Reset form
setName("");
- setSelectedModel("");
+ setSelectedModelId("");
},
onError: (error: any) => {
toast({
@@ -65,17 +70,16 @@ export function CreateAgentDialog({ open, onOpenChange }: CreateAgentDialogProps
});
const handleCreate = () => {
- const modelId = getModelIdByValue(selectedModel);
- if (!modelId) {
+ if (!selectedModelId) {
toast({
title: "Erro",
- description: "Modelo de IA inválido.",
+ description: "Selecione um modelo de IA.",
variant: "destructive",
});
return;
}
- createAgentMutation.mutate({ name, modelId });
+ createAgentMutation.mutate({ name, modelId: parseInt(selectedModelId) });
};
return (
@@ -101,13 +105,13 @@ export function CreateAgentDialog({ open, onOpenChange }: CreateAgentDialogProps
-