kaikaidai commited on
Commit
35e8314
·
verified ·
1 Parent(s): 04115ca

Create utils.py

Browse files
Files changed (1) hide show
  1. utils.py +27 -0
utils.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from dataclasses import dataclass
2
+ from datetime import datetime
3
+ import logging
4
+
5
+
6
+ def get_logger(sink_name: str = "core_utils") -> logging.Logger:
7
+ logging.basicConfig(
8
+ format="%(asctime)s,%(msecs)03d %(levelname)-8s "
9
+ "[%(filename)s:%(lineno)d] %(message)s",
10
+ datefmt="%Y-%m-%d:%H:%M:%S",
11
+ level=logging.INFO,
12
+ force=True,
13
+ )
14
+ logger = logging.getLogger(sink_name)
15
+ return logger
16
+
17
+
18
+ @dataclass
19
+ class Vote:
20
+ timestamp: str
21
+ prompt: str
22
+ response_a: str
23
+ response_b: str
24
+ model_a: str
25
+ model_b: str
26
+ winner: str
27
+ judge_id: str