@@ -89,6 +89,7 @@ def __post_init__(self):
89
89
if self .provider is None and self .name is not None :
90
90
self .provider = f"{ self .name } ({ self .id } )"
91
91
92
+
92
93
def to_dict (self ) -> Dict [str , Any ]:
93
94
fields = {"name" , "id" , "provider" }
94
95
_dict = {field : getattr (self , field ) for field in fields if getattr (self , field ) is not None }
@@ -161,6 +162,7 @@ def response(self, messages: List[Message]) -> ModelResponse:
161
162
ModelResponse: The model's response
162
163
"""
163
164
logger .debug (f"---------- { self .get_provider ()} Response Start ----------" )
165
+ logger .debug (f"---------- Model: { self .id } ----------" )
164
166
self ._log_messages (messages )
165
167
model_response = ModelResponse ()
166
168
@@ -223,6 +225,7 @@ async def aresponse(self, messages: List[Message]) -> ModelResponse:
223
225
ModelResponse: The model's response
224
226
"""
225
227
logger .debug (f"---------- { self .get_provider ()} Async Response Start ----------" )
228
+ logger .debug (f"---------- Model: { self .id } ----------" )
226
229
self ._log_messages (messages )
227
230
model_response = ModelResponse ()
228
231
@@ -460,6 +463,7 @@ def response_stream(self, messages: List[Message]) -> Iterator[ModelResponse]:
460
463
Iterator[ModelResponse]: Iterator of model responses
461
464
"""
462
465
logger .debug (f"---------- { self .get_provider ()} Response Stream Start ----------" )
466
+ logger .debug (f"---------- Model: { self .id } ----------" )
463
467
self ._log_messages (messages )
464
468
465
469
while True :
@@ -555,6 +559,7 @@ async def aresponse_stream(self, messages: List[Message]) -> AsyncIterator[Model
555
559
AsyncIterator[ModelResponse]: Async iterator of model responses
556
560
"""
557
561
logger .debug (f"---------- { self .get_provider ()} Async Response Stream Start ----------" )
562
+ logger .debug (f"---------- Model: { self .id } ----------" )
558
563
self ._log_messages (messages )
559
564
560
565
while True :
0 commit comments