11from abc import ABC , abstractmethod
22from dataclasses import dataclass
3- from typing import Any , Dict , List , Optional , Union
4-
5- from polywrap_core import Env , Uri , UriPackage , UriWrapper
3+ from polywrap_core import Uri , IUriResolver , Env , UriWrapper , UriPackage
4+ from polywrap_uri_resolvers import IUriResolver
5+ from typing import Dict , Any , List , Optional , Union
66from polywrap_uri_resolvers import UriResolverLike
77
88# from polywrap_plugin import PluginPackage
@@ -132,7 +132,6 @@ def add_package(self, uri_package: UriPackage):
132132 """
133133 self .config .packages .append (uri_package )
134134 return self
135-
136135 def add_packages (self , uri_packages : List [UriPackage ]):
137136 """
138137 Adds a list of packages to the list of packages
@@ -148,7 +147,7 @@ def remove_package(self, uri_package: UriPackage):
148147 self .config .packages .remove (uri_package )
149148 return self
150149
151- def set_resolver (self , uri_resolver ):
150+ def set_resolver (self , uri_resolver : UriResolverLike ):
152151 """
153152 Sets a single resolver for the `ClientConfig` object before it is built
154153 """
@@ -182,7 +181,6 @@ def build(self) -> ClientConfig:
182181 Returns a sanitized config object from the builder's config.
183182 """
184183 return self .config
185-
186184 def add (self , new_config : ClientConfig ) -> ClientConfig :
187185 """
188186 Returns a sanitized config object from the builder's config after receiving a partial `ClientConfig` object.
0 commit comments