diff --git a/src/backend/common/config/app_config.py b/src/backend/common/config/app_config.py
index 94b9772f0..5626752c1 100644
--- a/src/backend/common/config/app_config.py
+++ b/src/backend/common/config/app_config.py
@@ -7,7 +7,7 @@
 from azure.cosmos import CosmosClient
 from azure.identity import DefaultAzureCredential, ManagedIdentityCredential
 from dotenv import load_dotenv
-from agent_framework import Kernel
+from semantic_kernel import Kernel
 
 # Load environment variables from .env file
 load_dotenv()
diff --git a/src/backend/common/models/messages_kernel.py b/src/backend/common/models/messages_kernel.py
index f1fd2469e..03948b7b0 100644
--- a/src/backend/common/models/messages_kernel.py
+++ b/src/backend/common/models/messages_kernel.py
@@ -2,8 +2,8 @@
 from datetime import datetime, timezone
 from enum import Enum
 from typing import Any, Dict, List, Literal, Optional
-from pydantic import BaseModel, Field
 
+from semantic_kernel.kernel_pydantic import Field, KernelBaseModel
 
 
 class DataType(str, Enum):
@@ -75,7 +75,7 @@ class MessageRole(str, Enum):
     function = "function"
 
 
-class BaseDataModel(BaseModel):
+class BaseDataModel(KernelBaseModel):
     """Base data model with common fields."""
 
     id: str = Field(default_factory=lambda: str(uuid.uuid4()))
@@ -155,7 +155,7 @@ class TeamSelectionRequest(BaseDataModel):
     team_id: str
 
 
-class TeamAgent(BaseModel):
+class TeamAgent(KernelBaseModel):
     """Represents an agent within a team."""
 
     input_key: str
@@ -173,7 +173,7 @@ class TeamAgent(BaseModel):
     coding_tools: bool = False
 
 
-class StartingTask(BaseModel):
+class StartingTask(KernelBaseModel):
     """Represents a starting task for a team."""
 
     id: str
@@ -245,7 +245,7 @@ def update_step_counts(self):
 
 
 # Message classes for communication between agents
-class InputTask(BaseModel):
+class InputTask(KernelBaseModel):
     """Message representing the initial input task from the user."""
 
     session_id: str
@@ -253,7 +253,7 @@ class InputTask(BaseModel):
     # team_id: str
 
 
-class UserLanguage(BaseModel):
+class UserLanguage(KernelBaseModel):
     language: str
 
 
@@ -275,4 +275,4 @@ class AgentMessageData(BaseDataModel):
     content: str
     raw_data: str
     steps: List[Any] = Field(default_factory=list)
-    next_steps: List[Any] = Field(default_factory=list)
+    next_steps: List[Any] = Field(default_factory=list)
\ No newline at end of file
diff --git a/src/backend/common/utils/utils_kernel.py b/src/backend/common/utils/utils_kernel.py
index 380f94f79..85fc2fac0 100644
--- a/src/backend/common/utils/utils_kernel.py
+++ b/src/backend/common/utils/utils_kernel.py
@@ -4,7 +4,7 @@
 from typing import Any, Dict
 
 # Import agent factory and the new AppConfig
-from agent_framework.azure import AzureAIAgentClient
+from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
 from v3.magentic_agents.foundry_agent import FoundryAgentTemplate
 
 from v3.config.agent_registry import agent_registry