From 5606bfeea048f0a9bd0634a1533fd5ee93502b20 Mon Sep 17 00:00:00 2001
From: Itay Verkh <iverkh@lightricks.com>
Date: Thu, 19 Dec 2024 10:56:34 +0000
Subject: [PATCH] add `response_model` key to LM history entry

---
 dspy/clients/lm.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/dspy/clients/lm.py b/dspy/clients/lm.py
index 349e653396..af63e3db7a 100644
--- a/dspy/clients/lm.py
+++ b/dspy/clients/lm.py
@@ -110,6 +110,7 @@ def __call__(self, prompt=None, messages=None, **kwargs):
             timestamp=datetime.now().isoformat(),
             uuid=str(uuid.uuid4()),
             model=self.model,
+            response_model=response["model"],
             model_type=self.model_type,
         )
         self.history.append(entry)