Update mteb_eval_openai.py
Browse files- mteb_eval_openai.py +6 -0
mteb_eval_openai.py
CHANGED
|
@@ -5,6 +5,12 @@ import hashlib
|
|
| 5 |
import numpy as np
|
| 6 |
import requests
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
OPENAI_BASE_URL = os.environ.get('OPENAI_BASE_URL', '')
|
| 10 |
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY', '')
|
|
|
|
| 5 |
import numpy as np
|
| 6 |
import requests
|
| 7 |
|
| 8 |
+
import logging
|
| 9 |
+
import functools
|
| 10 |
+
from mteb import MTEB
|
| 11 |
+
from sentence_transformers import SentenceTransformer
|
| 12 |
+
logging.basicConfig(level=logging.INFO)
|
| 13 |
+
logger = logging.getLogger("main")
|
| 14 |
|
| 15 |
OPENAI_BASE_URL = os.environ.get('OPENAI_BASE_URL', '')
|
| 16 |
OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY', '')
|