Skip to content

Commit 185f279

Browse files
committed
fix: Separate module for shared code
1 parent 6541e00 commit 185f279

20 files changed

Lines changed: 19 additions & 19 deletions

adk/agenticlayer/adk/agent.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88
import httpx
99
from a2a.client import A2ACardResolver
1010
from a2a.utils.constants import AGENT_CARD_WELL_KNOWN_PATH
11-
from agenticlayer.config import InteractionType, McpTool, SubAgent
12-
from agenticlayer.constants import HTTP_HEADERS_SESSION_KEY
11+
from agenticlayer.shared.config import InteractionType, McpTool, SubAgent
12+
from agenticlayer.shared.constants import HTTP_HEADERS_SESSION_KEY
1313
from google.adk.agents import BaseAgent, LlmAgent
1414
from google.adk.agents.llm_agent import ToolUnion
1515
from google.adk.agents.readonly_context import ReadonlyContext

adk/agenticlayer/adk/agent_to_a2a.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313
from a2a.server.tasks import InMemoryTaskStore
1414
from a2a.types import AgentCapabilities, AgentCard
1515
from a2a.utils.constants import AGENT_CARD_WELL_KNOWN_PATH
16-
from agenticlayer.config import McpTool, SubAgent
17-
from agenticlayer.constants import HTTP_HEADERS_SESSION_KEY
16+
from agenticlayer.shared.config import McpTool, SubAgent
17+
from agenticlayer.shared.constants import HTTP_HEADERS_SESSION_KEY
1818
from google.adk.a2a.converters.request_converter import AgentRunRequest
1919
from google.adk.a2a.executor.a2a_agent_executor import A2aAgentExecutor
2020
from google.adk.agents import LlmAgent
@@ -224,7 +224,7 @@ async def lifespan(app: Starlette) -> AsyncIterator[None]:
224224
starlette_app = Starlette(lifespan=lifespan)
225225

226226
# Instrument the Starlette app with OpenTelemetry
227-
from agenticlayer.otel_starlette import instrument_starlette_app
227+
from agenticlayer.shared.otel_starlette import instrument_starlette_app
228228

229229
instrument_starlette_app(starlette_app)
230230

adk/agenticlayer/adk/otel.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
"""OpenTelemetry setup for a Google ADK Agent App."""
22

3-
from agenticlayer._otel import (
3+
from agenticlayer.shared.otel import (
44
_decode_body,
55
_is_text_content,
66
request_hook,
77
response_hook,
88
)
9-
from agenticlayer._otel import (
9+
from agenticlayer.shared.otel import (
1010
setup_otel as _setup_otel_base,
1111
)
1212

adk/tests/fixtures/app_factory.py

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

77
import pytest_asyncio
8-
from agenticlayer.config import McpTool, SubAgent
8+
from agenticlayer.shared.config import McpTool, SubAgent
99
from asgi_lifespan import LifespanManager
1010
from google.adk.agents import LlmAgent
1111
from httpx_retries import Retry

adk/tests/test_agent_integration.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,8 @@
66
import pytest
77
import respx
88
from a2a.client.errors import A2AClientHTTPError
9-
from agenticlayer.config import InteractionType, McpTool, SubAgent
10-
from agenticlayer.loguru_config import setup_logging
9+
from agenticlayer.shared.config import InteractionType, McpTool, SubAgent
10+
from agenticlayer.shared.loguru_config import setup_logging
1111
from asgi_lifespan import LifespanManager
1212
from fastmcp import Context, FastMCP
1313
from httpx_retries import Retry

adk/tests/test_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
"""Unit tests for config parsing functions."""
22

33
import pytest
4-
from agenticlayer.config import InteractionType, parse_sub_agents, parse_tools
4+
from agenticlayer.shared.config import InteractionType, parse_sub_agents, parse_tools
55

66

77
class TestParseSubAgents:

msaf/agenticlayer/msaf/agent.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from a2a.utils.constants import AGENT_CARD_WELL_KNOWN_PATH
1414
from agent_framework._mcp import MCPStreamableHTTPTool
1515
from agent_framework._tools import FunctionTool
16-
from agenticlayer.config import McpTool, SubAgent
16+
from agenticlayer.shared.config import McpTool, SubAgent
1717
from httpx_retries import Retry, RetryTransport
1818

1919
logger = logging.getLogger(__name__)

msaf/agenticlayer/msaf/agent_to_a2a.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
from agent_framework import SupportsAgentRun
3030
from agent_framework._mcp import MCPStreamableHTTPTool
3131
from agent_framework._tools import FunctionTool
32-
from agenticlayer.config import McpTool, SubAgent
32+
from agenticlayer.shared.config import McpTool, SubAgent
3333
from httpx_retries import Retry
3434
from starlette.applications import Starlette
3535

@@ -298,7 +298,7 @@ async def lifespan(app: Starlette) -> AsyncIterator[None]:
298298
starlette_app = Starlette(lifespan=lifespan)
299299

300300
# Instrument the Starlette app with OpenTelemetry
301-
from agenticlayer.otel_starlette import instrument_starlette_app
301+
from agenticlayer.shared.otel_starlette import instrument_starlette_app
302302

303303
instrument_starlette_app(starlette_app)
304304

msaf/agenticlayer/msaf/otel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
"""OpenTelemetry setup for a Microsoft Agent Framework Agent App."""
22

3-
from agenticlayer._otel import setup_otel as _setup_otel_shared
3+
from agenticlayer.shared.otel import setup_otel as _setup_otel_shared
44

55
__all__ = ["setup_otel"]
66

msaf/pyproject.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ requires-python = ">=3.14,<3.15"
66
dependencies = [
77
"agentic-layer-sdk",
88
"agent-framework-core>=1.0.0rc2",
9-
# "httpx",
109
"httpx-retries>=0.4.5",
1110
]
1211

0 commit comments

Comments
 (0)