Skip to content

Commit 4d404b6

Browse files
authored
fix(server): check registry cron validation error (#1006)
Signed-off-by: Radek Ježek <radek.jezek@ibm.com>
1 parent 0e0bdab commit 4d404b6

File tree

1 file changed

+3
-4
lines changed
  • apps/beeai-server/src/beeai_server/jobs/crons

1 file changed

+3
-4
lines changed

apps/beeai-server/src/beeai_server/jobs/crons/provider.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,11 @@
99
import httpx
1010
from kink import inject
1111
from procrastinate import Blueprint
12+
from pydantic import RootModel
1213

1314
from beeai_server import get_configuration
1415
from beeai_server.configuration import Configuration
15-
from beeai_server.domain.models.provider import Provider
16+
from beeai_server.domain.models.provider import ProviderLocation
1617
from beeai_server.exceptions import EntityNotFoundError
1718
from beeai_server.service_layer.services.provider import ProviderService
1819
from beeai_server.service_layer.unit_of_work import IUnitOfWorkFactory
@@ -45,9 +46,7 @@ async def check_registry(timestamp: int, configuration: Configuration, provider_
4546
for registry in configuration.agent_registry.locations.values():
4647
for provider_location in await registry.load():
4748
try:
48-
provider_id = Provider(
49-
source=provider_location, env=[]
50-
).id # dummy object to calculate ID from location
49+
provider_id = RootModel[ProviderLocation](root=provider_location).root.provider_id
5150
desired_providers[provider_id] = provider_location
5251
registry_by_provider_id[provider_id] = registry
5352
except ValueError as e:

0 commit comments

Comments
 (0)