fakeym commited on
Commit
bf85061
1 Parent(s): bebf9f7

Update travel/self_rag_tool.py

Browse files
Files changed (1) hide show
  1. 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
  将用户查询路由到最相关的数据源