From efbb2d1416fda2e1f7c390dbcc7eac4f79d88e0a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Christian=20Ladr=C3=B3n=20De=20Guevara?=
 <christianlgr@gmail.com>
Date: Wed, 13 Mar 2024 21:06:22 -0700
Subject: [PATCH] Add extra_body field to LLMConfig dataclass for Azure AI
 Search support (#2011)

Co-authored-by: Chi Wang <wang.chi@microsoft.com>
---
 samples/apps/autogen-studio/autogenstudio/datamodel.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/samples/apps/autogen-studio/autogenstudio/datamodel.py b/samples/apps/autogen-studio/autogenstudio/datamodel.py
index 1068aa281f25..cd8d5a40f6eb 100644
--- a/samples/apps/autogen-studio/autogenstudio/datamodel.py
+++ b/samples/apps/autogen-studio/autogenstudio/datamodel.py
@@ -93,6 +93,7 @@ class LLMConfig:
     cache_seed: Optional[Union[int, None]] = None
     timeout: Optional[int] = None
     max_tokens: Optional[int] = None
+    extra_body: Optional[dict] = None
 
     def dict(self):
         result = asdict(self)