From 869001fdebda004aecd79aab7abf7acdaf41ef15 Mon Sep 17 00:00:00 2001 From: Diogo Sousa Date: Mon, 17 Jul 2023 12:49:17 +0100 Subject: [PATCH] allow provider default none --- openbb_sdk/sdk/core/openbb_core/app/command_runner.py | 1 + openbb_sdk/sdk/core/openbb_core/app/model/defaults.py | 4 ++-- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/openbb_sdk/sdk/core/openbb_core/app/command_runner.py b/openbb_sdk/sdk/core/openbb_core/app/command_runner.py index d75074bcbfb..82bbb716425 100644 --- a/openbb_sdk/sdk/core/openbb_core/app/command_runner.py +++ b/openbb_sdk/sdk/core/openbb_core/app/command_runner.py @@ -150,6 +150,7 @@ class ParametersBuilder: provider = ( random_choice if route_defaults is None + or route_defaults.get("provider", None) is None else route_defaults.get("provider", random_choice) ) kwargs["provider_choices"] = {"provider": provider} diff --git a/openbb_sdk/sdk/core/openbb_core/app/model/defaults.py b/openbb_sdk/sdk/core/openbb_core/app/model/defaults.py index 8170a1a31c7..85f4a7058f8 100644 --- a/openbb_sdk/sdk/core/openbb_core/app/model/defaults.py +++ b/openbb_sdk/sdk/core/openbb_core/app/model/defaults.py @@ -1,4 +1,4 @@ -from typing import Dict +from typing import Dict, Optional from pydantic import BaseModel, Field @@ -9,7 +9,7 @@ class Defaults(BaseModel): class Config: validate_assignment = True - routes: Dict[str, Dict[str, str]] = Field(default_factory=dict) + routes: Dict[str, Dict[str, Optional[str]]] = Field(default_factory=dict) def __repr__(self) -> str: return (