Skip to content

Commit e7955ef

Browse files
authored
Revert "add lm callback back (#8169)" (#8201)
This reverts commit ffb415f.
1 parent ffb415f commit e7955ef

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

dspy/clients/lm.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from dspy.clients.provider import Provider, TrainingJob
1515
from dspy.clients.utils_finetune import TrainDataFormat
1616
from dspy.dsp.utils.settings import settings
17-
from dspy.utils.callback import BaseCallback, with_callbacks
17+
from dspy.utils.callback import BaseCallback
1818

1919
from .base_lm import BaseLM
2020

@@ -114,7 +114,6 @@ def _get_cached_completion_fn(self, completion_fn, cache, enable_memory_cache):
114114

115115
return completion_fn, litellm_cache_args
116116

117-
@with_callbacks
118117
def forward(self, prompt=None, messages=None, **kwargs):
119118
# Build the request.
120119
cache = kwargs.pop("cache", self.cache)
@@ -136,7 +135,6 @@ def forward(self, prompt=None, messages=None, **kwargs):
136135
settings.usage_tracker.add_usage(self.model, dict(results.usage))
137136
return results
138137

139-
@with_callbacks
140138
async def aforward(self, prompt=None, messages=None, **kwargs):
141139
# Build the request.
142140
cache = kwargs.pop("cache", self.cache)

0 commit comments

Comments
 (0)