@@ -147,12 +147,12 @@ async def _a_generate_reason(self, input: str):
147147 score = format (self .score , ".2f" ),
148148 )
149149 if self .using_native_model :
150- res , cost = await self .model .a_generate (prompt , schema = Reason )
150+ res , cost = await self .model .a_generate (prompt , schema = ContextualRelevancyScoreReason )
151151 self .evaluation_cost += cost
152152 return res .reason
153153 else :
154154 try :
155- res : Reason = await self .model .a_generate (prompt , schema = Reason )
155+ res : ContextualRelevancyScoreReason = await self .model .a_generate (prompt , schema = ContextualRelevancyScoreReason )
156156 return res .reason
157157 except TypeError :
158158 res = await self .model .a_generate (prompt )
@@ -179,12 +179,12 @@ def _generate_reason(self, input: str):
179179 score = format (self .score , ".2f" ),
180180 )
181181 if self .using_native_model :
182- res , cost = self .model .generate (prompt , schema = Reason )
182+ res , cost = self .model .generate (prompt , schema = ContextualRelevancyScoreReason )
183183 self .evaluation_cost += cost
184184 return res .reason
185185 else :
186186 try :
187- res : Reason = self .model .generate (prompt , schema = Reason )
187+ res : ContextualRelevancyScoreReason = self .model .generate (prompt , schema = ContextualRelevancyScoreReason )
188188 return res .reason
189189 except TypeError :
190190 res = self .model .generate (prompt )
0 commit comments