Add Support for Multiple LLMs and Frameworks

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()}")

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

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"