Skip to content

Commit 0365e2f

Browse files
authored
Merge pull request microsoft#573 from Fr4nc3/macae-rfp-af-101725
Macae rfp af 101725
2 parents f69edb1 + a47b0a2 commit 0365e2f

File tree

3 files changed

+9
-9
lines changed

3 files changed

+9
-9
lines changed

src/backend/common/config/app_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from azure.cosmos import CosmosClient
88
from azure.identity import DefaultAzureCredential, ManagedIdentityCredential
99
from dotenv import load_dotenv
10-
from agent_framework import Kernel
10+
from semantic_kernel import Kernel
1111

1212
# Load environment variables from .env file
1313
load_dotenv()

src/backend/common/models/messages_kernel.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,8 @@
22
from datetime import datetime, timezone
33
from enum import Enum
44
from typing import Any, Dict, List, Literal, Optional
5-
from pydantic import BaseModel, Field
65

6+
from semantic_kernel.kernel_pydantic import Field, KernelBaseModel
77

88

99
class DataType(str, Enum):
@@ -75,7 +75,7 @@ class MessageRole(str, Enum):
7575
function = "function"
7676

7777

78-
class BaseDataModel(BaseModel):
78+
class BaseDataModel(KernelBaseModel):
7979
"""Base data model with common fields."""
8080

8181
id: str = Field(default_factory=lambda: str(uuid.uuid4()))
@@ -155,7 +155,7 @@ class TeamSelectionRequest(BaseDataModel):
155155
team_id: str
156156

157157

158-
class TeamAgent(BaseModel):
158+
class TeamAgent(KernelBaseModel):
159159
"""Represents an agent within a team."""
160160

161161
input_key: str
@@ -173,7 +173,7 @@ class TeamAgent(BaseModel):
173173
coding_tools: bool = False
174174

175175

176-
class StartingTask(BaseModel):
176+
class StartingTask(KernelBaseModel):
177177
"""Represents a starting task for a team."""
178178

179179
id: str
@@ -245,15 +245,15 @@ def update_step_counts(self):
245245

246246

247247
# Message classes for communication between agents
248-
class InputTask(BaseModel):
248+
class InputTask(KernelBaseModel):
249249
"""Message representing the initial input task from the user."""
250250

251251
session_id: str
252252
description: str # Initial goal
253253
# team_id: str
254254

255255

256-
class UserLanguage(BaseModel):
256+
class UserLanguage(KernelBaseModel):
257257
language: str
258258

259259

@@ -275,4 +275,4 @@ class AgentMessageData(BaseDataModel):
275275
content: str
276276
raw_data: str
277277
steps: List[Any] = Field(default_factory=list)
278-
next_steps: List[Any] = Field(default_factory=list)
278+
next_steps: List[Any] = Field(default_factory=list)

src/backend/common/utils/utils_kernel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from typing import Any, Dict
55

66
# Import agent factory and the new AppConfig
7-
from agent_framework.azure import AzureAIAgentClient
7+
from semantic_kernel.agents.azure_ai.azure_ai_agent import AzureAIAgent
88
from v3.magentic_agents.foundry_agent import FoundryAgentTemplate
99

1010
from v3.config.agent_registry import agent_registry

0 commit comments

Comments
 (0)