Partial cleanup of weaviate

This commit is contained in:
Zain Hasan 2024-10-07 01:21:50 -04:00 committed by Ashwin Bharambe
parent 862f8ddb8d
commit 118c0ef105
4 changed files with 82 additions and 103 deletions

View file

@ -39,14 +39,6 @@ RoutedProtocol = Union[
]
class ModelRegistry(Protocol):
def get_model(self, identifier: str) -> ModelDef: ...
class MemoryBankRegistry(Protocol):
def get_memory_bank(self, identifier: str) -> MemoryBankDef: ...
# Example: /inference, /safety
class AutoRoutedProviderSpec(ProviderSpec):
provider_type: str = "router"

View file

@ -64,7 +64,6 @@ async def resolve_impls_with_routing(run_config: StackRunConfig) -> Dict[Api, An
+ [x.value for x in routing_table_apis]
+ [x.value for x in router_apis]
)
print(f"{apis_to_serve=}")
for info in builtin_automatically_routed_apis():
if info.router_api.value not in apis_to_serve: