From 9c1565e3dbf2e7e8c4b8973123fa57f100e18d5f Mon Sep 17 00:00:00 2001
From: divyanshu-iitian <23f2001496@ds.study.iitm.ac.in>
Date: Mon, 16 Feb 2026 11:12:28 +0530
Subject: [PATCH] feat: add multi-LLM support

---
 burr/system.py    | 18 ++++++++++++++++++
 burr/telemetry.py |  2 +-
 setup.cfg         |  7 +++++++
 3 files changed, 26 insertions(+), 1 deletion(-)

diff --git a/burr/system.py b/burr/system.py
index 75a45f089..65527556b 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 791b8504a..1f8f28092 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 60ee3500d..9cf0dd181 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