Skip to content

Commit 621a8a7

Browse files
rootroot
authored andcommitted
import fixes
1 parent 5a9691a commit 621a8a7

File tree

2 files changed

+5
-20
lines changed

2 files changed

+5
-20
lines changed

sdk/agentserver/azure-ai-agentserver-githubcopilot/azure/ai/agentserver/githubcopilot/_copilot_adapter.py

Lines changed: 4 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -27,15 +27,8 @@
2727
from copilot import CopilotClient
2828
from copilot.generated.session_events import SessionEventType
2929

30-
# These types move between SDK versions/platforms. Try multiple paths.
31-
try:
32-
from copilot import PermissionRequestResult, ProviderConfig
33-
except ImportError:
34-
try:
35-
from copilot.types import PermissionRequestResult, ProviderConfig
36-
except ImportError:
37-
PermissionRequestResult = None
38-
ProviderConfig = dict
30+
# Import types from copilot SDK
31+
from copilot.session import PermissionRequestResult, ProviderConfig
3932

4033
from azure.ai.agentserver.core.constants import Constants
4134
from azure.ai.agentserver.core.logger import get_logger
@@ -302,9 +295,7 @@ async def agent_run(
302295
acl = self._acl
303296

304297
def _perm_result(**kwargs):
305-
if PermissionRequestResult is not None:
306-
return PermissionRequestResult(**kwargs)
307-
return kwargs
298+
return PermissionRequestResult(**kwargs)
308299

309300
def _on_permission(req, _ctx):
310301
kind = getattr(req, "kind", "unknown")
@@ -901,9 +892,7 @@ async def agent_run(self, context: AgentRunContext):
901892
acl = self._acl
902893

903894
def _perm_result_boot(**kwargs):
904-
if PermissionRequestResult is not None:
905-
return PermissionRequestResult(**kwargs)
906-
return kwargs
895+
return PermissionRequestResult(**kwargs)
907896

908897
def _on_permission_boot(req, _ctx):
909898
kind = getattr(req, "kind", "unknown")

sdk/agentserver/azure-ai-agentserver-githubcopilot/azure/ai/agentserver/githubcopilot/_tool_discovery.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -22,11 +22,7 @@
2222

2323
import yaml
2424

25-
try:
26-
from copilot import Tool
27-
except ImportError:
28-
# Copilot SDK renamed/moved Tool in some versions
29-
Tool = None # type: ignore
25+
from copilot.tools import Tool
3026

3127
logger = logging.getLogger(__name__)
3228

0 commit comments

Comments
 (0)