Update travel/self_rag_tool.py
Browse files- travel/self_rag_tool.py +2 -1
travel/self_rag_tool.py
CHANGED
@@ -11,7 +11,8 @@ from typing import Literal
|
|
11 |
from transformers import AutoModel, AutoTokenizer
|
12 |
from pypinyin import lazy_pinyin
|
13 |
import json
|
14 |
-
|
|
|
15 |
class LocationType(BaseModel):
|
16 |
"""
|
17 |
将用户查询路由到最相关的数据源
|
|
|
11 |
from transformers import AutoModel, AutoTokenizer
|
12 |
from pypinyin import lazy_pinyin
|
13 |
import json
|
14 |
+
from dotenv import load_dotenv
|
15 |
+
_ = load_dotenv()
|
16 |
class LocationType(BaseModel):
|
17 |
"""
|
18 |
将用户查询路由到最相关的数据源
|