File size: 754 Bytes
e773d98 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
from typing import Optional, List, Dict, Any
class RankingExecInfo:
def __init__(
self, prompt, response: str, input_token_count: int, output_token_count: int
):
self.prompt = prompt
self.response = response
self.input_token_count = input_token_count
self.output_token_count = output_token_count
def __repr__(self):
return str(self.__dict__)
class Result:
def __init__(
self,
query: str,
hits: List[Dict[str, Any]],
ranking_exec_summary: Optional[List[RankingExecInfo]] = None,
):
self.query = query
self.hits = hits
self.ranking_exec_summary = ranking_exec_summary
def __repr__(self):
return str(self.__dict__)
|