github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-31 15:47:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

cfg: update

Browse Source
This commit is contained in:
csunny 2023-05-18 17:32:53 +08:00
parent 9ddd089381
commit 0d370a1a64
3 changed files with 2 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/__init__.py
View File

@ -1,11 +1,5 @@
from pilot.source_embedding import (SourceEmbedding, register) from pilot.source_embedding import (SourceEmbedding, register)
import os
import sys
ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.append(ROOT_PATH)
__all__ = [ __all__ = [
"SourceEmbedding", "SourceEmbedding",
"register" "register"

7
pilot/configs/model_config.py
View File

@ -28,12 +28,5 @@ ISDEBUG = False
VECTOR_SEARCH_TOP_K = 3 VECTOR_SEARCH_TOP_K = 3
# LLM_MODEL = "vicuna-13b"
# LIMIT_MODEL_CONCURRENCY = 5
# MAX_POSITION_EMBEDDINGS = 4096
# VICUNA_MODEL_SERVER = "http://121.41.167.183:8000"
VS_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "vs_store") VS_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "vs_store")
KNOWLEDGE_UPLOAD_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "knowledge") KNOWLEDGE_UPLOAD_ROOT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)), "knowledge")

4
pilot/server/webserver.py
View File

@ -6,6 +6,7 @@ import os
import shutil import shutil
import uuid import uuid
import json import json
import sys
import time import time
import gradio as gr import gradio as gr
import datetime import datetime
@ -13,8 +14,7 @@ import requests
from urllib.parse import urljoin from urllib.parse import urljoin
from langchain import PromptTemplate from langchain import PromptTemplate
import os
import sys
ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))) ROOT_PATH = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
sys.path.append(ROOT_PATH) sys.path.append(ROOT_PATH)