Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion src/backend/common/config/app_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -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()
Expand Down
14 changes: 7 additions & 7 deletions src/backend/common/models/messages_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down Expand Up @@ -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()))
Expand Down Expand Up @@ -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
Expand All @@ -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
Expand Down Expand Up @@ -245,15 +245,15 @@ 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
description: str # Initial goal
# team_id: str


class UserLanguage(BaseModel):
class UserLanguage(KernelBaseModel):
language: str


Expand All @@ -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)
2 changes: 1 addition & 1 deletion src/backend/common/utils/utils_kernel.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Loading