Germano Cavalcante
commited on
Commit
·
3e9f56d
1
Parent(s):
e8fa699
Fixes
Browse files- config.py +2 -2
- routers/tool_find_related.py +2 -2
- routers/tool_wiki_search.py +2 -2
config.py
CHANGED
@@ -3,8 +3,8 @@ import os
|
|
3 |
|
4 |
|
5 |
class Settings(BaseSettings):
|
6 |
-
huggingface_key: str = os.environ.get("huggingface_key")
|
7 |
-
OPENAI_API_KEY: str = os.environ.get("OPENAI_API_KEY")
|
8 |
embedding_api: str = "sbert"
|
9 |
embedding_model: str = "mano-wii/BAAI_bge-base-en-v1.5-tunned-for-blender-issues"
|
10 |
# embedding_api: str = "openai"
|
|
|
3 |
|
4 |
|
5 |
class Settings(BaseSettings):
|
6 |
+
huggingface_key: str = os.environ.get("huggingface_key", "")
|
7 |
+
OPENAI_API_KEY: str = os.environ.get("OPENAI_API_KEY", "")
|
8 |
embedding_api: str = "sbert"
|
9 |
embedding_model: str = "mano-wii/BAAI_bge-base-en-v1.5-tunned-for-blender-issues"
|
10 |
# embedding_api: str = "openai"
|
routers/tool_find_related.py
CHANGED
@@ -13,8 +13,8 @@ from fastapi import APIRouter
|
|
13 |
from fastapi.responses import PlainTextResponse
|
14 |
|
15 |
try:
|
16 |
-
from .rag import EMBEDDING_CTX
|
17 |
-
from .utils_gitea import gitea_fetch_issues, gitea_json_issue_get, gitea_issues_body_updated_at_get
|
18 |
except:
|
19 |
from rag import EMBEDDING_CTX
|
20 |
from utils_gitea import gitea_fetch_issues, gitea_json_issue_get, gitea_issues_body_updated_at_get
|
|
|
13 |
from fastapi.responses import PlainTextResponse
|
14 |
|
15 |
try:
|
16 |
+
from routers.rag import EMBEDDING_CTX
|
17 |
+
from routers.utils_gitea import gitea_fetch_issues, gitea_json_issue_get, gitea_issues_body_updated_at_get
|
18 |
except:
|
19 |
from rag import EMBEDDING_CTX
|
20 |
from utils_gitea import gitea_fetch_issues, gitea_json_issue_get, gitea_issues_body_updated_at_get
|
routers/tool_wiki_search.py
CHANGED
@@ -13,8 +13,8 @@ from sentence_transformers import util
|
|
13 |
from typing import Dict, List, Tuple, Set, LiteralString
|
14 |
|
15 |
try:
|
16 |
-
from .rag import SplitDocs, EMBEDDING_CTX
|
17 |
-
from .utils_gitea import gitea_wiki_page_get, gitea_wiki_pages_get
|
18 |
except:
|
19 |
from rag import SplitDocs, EMBEDDING_CTX
|
20 |
from utils_gitea import gitea_wiki_page_get, gitea_wiki_pages_get
|
|
|
13 |
from typing import Dict, List, Tuple, Set, LiteralString
|
14 |
|
15 |
try:
|
16 |
+
from routers.rag import SplitDocs, EMBEDDING_CTX
|
17 |
+
from routers.utils_gitea import gitea_wiki_page_get, gitea_wiki_pages_get
|
18 |
except:
|
19 |
from rag import SplitDocs, EMBEDDING_CTX
|
20 |
from utils_gitea import gitea_wiki_page_get, gitea_wiki_pages_get
|