diff --git a/burr/system.py b/burr/system.py index 75a45f08..65527556 100644 --- a/burr/system.py +++ b/burr/system.py @@ -29,3 +29,21 @@ def now(): return datetime.datetime.now(utc) + + +# Added support for multiple LLMs and frameworks +class LLMFrameworkConfig: + def __init__(self, llm_name, framework_name): + self.llm_name = llm_name + self.framework_name = framework_name + + def get_llm_name(self): + return self.llm_name + + def get_framework_name(self): + return self.framework_name + + +# Example usage of LLMFrameworkConfig +llm_config = LLMFrameworkConfig("OpenAI GPT-3", "PyTorch") +print(f"LLM: {llm_config.get_llm_name()}, Framework: {llm_config.get_framework_name()}") \ No newline at end of file diff --git a/burr/telemetry.py b/burr/telemetry.py index 791b8504..1f8f2809 100644 --- a/burr/telemetry.py +++ b/burr/telemetry.py @@ -307,4 +307,4 @@ def wrapped_fn(*args, **kwargs): f"Failed to send telemetry for function usage. Encountered: {e}\n" ) - return wrapped_fn + return wrapped_fn \ No newline at end of file diff --git a/setup.cfg b/setup.cfg index 60ee3500..9cf0dd18 100644 --- a/setup.cfg +++ b/setup.cfg @@ -33,3 +33,10 @@ ignore = [isort] known_first_party=hamilton known_local_folder=tests + +# Added configuration options for multiple LLMs and frameworks +[burrr] +supported_llms = ["llm1", "llm2"] +default_llm = "llm1" +supported_frameworks = ["framework1", "framework2"] +default_framework = "framework1" \ No newline at end of file