From b36715d2b6b4a2c950d368a25f047bfd2649455b Mon Sep 17 00:00:00 2001
From: Saumya <saumyadas2017@gmail.com>
Date: Wed, 30 Jul 2025 23:37:36 +0530
Subject: [PATCH] fix agent creation

---
 backend/agent/api.py                          | 40 +++++++++----------
 .../agents/agent-version-switcher.tsx         | 14 +++----
 2 files changed, 24 insertions(+), 30 deletions(-)

diff --git a/backend/agent/api.py b/backend/agent/api.py
index f0428d36..009ad9c9 100644
--- a/backend/agent/api.py
+++ b/backend/agent/api.py
@@ -1623,42 +1623,38 @@ async def create_agent(
                 change_description="Initial version"
             )
             
-            agent['current_version_id'] = version['version_id']
+            agent['current_version_id'] = version.version_id
             agent['version_count'] = 1
-            
-            # Create proper AgentVersionResponse from version dict
+
             current_version = AgentVersionResponse(
-                version_id=version['version_id'],
-                agent_id=version['agent_id'],
-                version_number=version['version_number'],
-                version_name=version['version_name'],
-                system_prompt=version['system_prompt'],
-                configured_mcps=version.get('configured_mcps', []),
-                custom_mcps=version.get('custom_mcps', []),
-                agentpress_tools=version.get('agentpress_tools', {}),
-                is_active=version.get('is_active', True),
-                created_at=version['created_at'],
-                updated_at=version.get('updated_at', version['created_at']),
-                created_by=version.get('created_by')
+                version_id=version.version_id,
+                agent_id=version.agent_id,
+                version_number=version.version_number,
+                version_name=version.version_name,
+                system_prompt=version.system_prompt,
+                configured_mcps=version.configured_mcps,
+                custom_mcps=version.custom_mcps,
+                agentpress_tools=version.agentpress_tools,
+                is_active=version.is_active,
+                created_at=version.created_at.isoformat(),
+                updated_at=version.updated_at.isoformat(),
+                created_by=version.created_by
             )
         except Exception as e:
             logger.error(f"Error creating initial version: {str(e)}")
-            # Clean up the agent if version creation fails
             await client.table('agents').delete().eq('agent_id', agent['agent_id']).execute()
             raise HTTPException(status_code=500, detail="Failed to create initial version")
         
         logger.info(f"Created agent {agent['agent_id']} with v1 for user: {user_id}")
-        
-        # Use version data for the response
         return AgentResponse(
             agent_id=agent['agent_id'],
             account_id=agent['account_id'],
             name=agent['name'],
             description=agent.get('description'),
-            system_prompt=version['system_prompt'],
-            configured_mcps=version.get('configured_mcps', []),
-            custom_mcps=version.get('custom_mcps', []),
-            agentpress_tools=version.get('agentpress_tools', {}),
+            system_prompt=version.system_prompt,
+            configured_mcps=version.configured_mcps,
+            custom_mcps=version.custom_mcps,
+            agentpress_tools=version.agentpress_tools,
             is_default=agent.get('is_default', False),
             is_public=agent.get('is_public', False),
             tags=agent.get('tags', []),
diff --git a/frontend/src/components/agents/agent-version-switcher.tsx b/frontend/src/components/agents/agent-version-switcher.tsx
index 227b53f5..47515ad8 100644
--- a/frontend/src/components/agents/agent-version-switcher.tsx
+++ b/frontend/src/components/agents/agent-version-switcher.tsx
@@ -199,14 +199,12 @@ export function AgentVersionSwitcher({
             })}
           </div>
           {versions.length === 1 && (
-            <div className="p-2">
-              <Alert>
-                <AlertCircle className="h-4 w-4" />
-                <AlertDescription>
-                  This is the first version. Make changes to create a new version.
-                </AlertDescription>
-              </Alert>
-            </div>
+            <Alert>
+              <AlertCircle className="h-4 w-4" />
+              <AlertDescription>
+                This is the first version. Make changes to create a new version.
+              </AlertDescription>
+            </Alert>
           )}
         </DropdownMenuContent>
       </DropdownMenu>