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
42 changes: 22 additions & 20 deletions demo/mock_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,13 +9,17 @@
UiPathRuntimeStorageProtocol,
)

from demo.mock_greeting_runtime import (
ENTRYPOINT_GREETING,
MockGreetingRuntime,
from demo.mock_invoice_runtime import (
ENTRYPOINT_INVOICE,
MockInvoiceRuntime,
)
from demo.mock_numbers_runtime import (
ENTRYPOINT_ANALYZE_NUMBERS,
MockNumberAnalyticsRuntime,
from demo.mock_movies_runtime import (
ENTRYPOINT_MOVIES,
MockMoviesRuntime,
)
from demo.mock_pharma_runtime import (
ENTRYPOINT_PHARMA,
MockPharmaRuntime,
)
from demo.mock_support_runtime import (
ENTRYPOINT_SUPPORT_CHAT,
Expand All @@ -32,12 +36,7 @@
logger = logging.getLogger(__name__)

# Template mappings: entrypoint -> (events_file, schema_file)
TEMPLATE_RUNTIMES = {
"chat/movies.py:graph": (
"chat_agent/events.json",
"chat_agent/entry-points.json",
),
}
TEMPLATE_RUNTIMES: dict[str, tuple[str, str]] = {}


class MockRuntimeFactory:
Expand All @@ -51,14 +50,16 @@ async def new_runtime(
self, entrypoint: str, runtime_id: str, **kwargs
) -> UiPathRuntimeProtocol:
"""Create a new runtime instance for the given entrypoint."""
if entrypoint == ENTRYPOINT_GREETING:
return MockGreetingRuntime(entrypoint=entrypoint)
if entrypoint == ENTRYPOINT_ANALYZE_NUMBERS:
return MockNumberAnalyticsRuntime(entrypoint=entrypoint)
if entrypoint == ENTRYPOINT_PHARMA:
return MockPharmaRuntime(entrypoint=entrypoint)
if entrypoint == ENTRYPOINT_INVOICE:
return MockInvoiceRuntime(entrypoint=entrypoint)
if entrypoint == ENTRYPOINT_SUPPORT_CHAT:
return MockSupportChatRuntime(entrypoint=entrypoint)
if entrypoint == ENTRYPOINT_TELEMETRY:
return MockTelemetryRuntime(entrypoint=entrypoint)
if entrypoint == ENTRYPOINT_MOVIES:
return MockMoviesRuntime(entrypoint=entrypoint)

if entrypoint in TEMPLATE_RUNTIMES:
events_file, schema_file = TEMPLATE_RUNTIMES[entrypoint]
Expand All @@ -73,9 +74,9 @@ async def new_runtime(

# Fallback: still return something so the demo doesn't explode
logger.warning(
"Unknown entrypoint %r, falling back to GreetingRuntime", entrypoint
"Unknown entrypoint %r, falling back to PharmaRuntime", entrypoint
)
return MockGreetingRuntime(entrypoint=entrypoint)
return MockPharmaRuntime(entrypoint=entrypoint)

async def get_settings(self) -> UiPathRuntimeFactorySettings | None:
"""Return factory settings (no-op for mock)."""
Expand All @@ -89,9 +90,10 @@ def discover_entrypoints(self) -> list[str]:
"""Return all available entrypoints."""
return [
ENTRYPOINT_TELEMETRY,
ENTRYPOINT_GREETING,
ENTRYPOINT_ANALYZE_NUMBERS,
ENTRYPOINT_PHARMA,
ENTRYPOINT_INVOICE,
ENTRYPOINT_SUPPORT_CHAT,
ENTRYPOINT_MOVIES,
*TEMPLATE_RUNTIMES.keys(),
]

Expand Down
171 changes: 0 additions & 171 deletions demo/mock_greeting_runtime.py

This file was deleted.

Loading