68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
FORCE_SHOW_FIELDS = [
|
|
"allowed_tools",
|
|
"memory",
|
|
"prefix",
|
|
"examples",
|
|
"temperature",
|
|
"model_name",
|
|
"headers",
|
|
"max_value_length",
|
|
"max_tokens",
|
|
"google_cse_id",
|
|
]
|
|
|
|
DEFAULT_PROMPT = """
|
|
I want you to act as a naming consultant for new companies.
|
|
|
|
Here are some examples of good company names:
|
|
|
|
- search engine, Google
|
|
- social media, Facebook
|
|
- video sharing, YouTube
|
|
|
|
The name should be short, catchy and easy to remember.
|
|
|
|
What is a good name for a company that makes {product}?
|
|
"""
|
|
|
|
SYSTEM_PROMPT = """
|
|
You are a helpful assistant that talks casually about life in general.
|
|
You are a good listener and you can talk about anything.
|
|
"""
|
|
|
|
HUMAN_PROMPT = "{input}"
|
|
|
|
QA_CHAIN_TYPES = ["stuff", "map_reduce", "map_rerank", "refine"]
|
|
|
|
CTRANSFORMERS_DEFAULT_CONFIG = {
|
|
"top_k": 40,
|
|
"top_p": 0.95,
|
|
"temperature": 0.8,
|
|
"repetition_penalty": 1.1,
|
|
"last_n_tokens": 64,
|
|
"seed": -1,
|
|
"max_new_tokens": 256,
|
|
"stop": None,
|
|
"stream": False,
|
|
"reset": True,
|
|
"batch_size": 8,
|
|
"threads": -1,
|
|
"context_length": -1,
|
|
"gpu_layers": 0,
|
|
}
|
|
|
|
# This variable is used to tell the user
|
|
# that it can be changed to use other APIs
|
|
# like Prem and LocalAI
|
|
OPENAI_API_BASE_INFO = """
|
|
The base URL of the OpenAI API. Defaults to https://api.openai.com/v1.
|
|
|
|
You can change this to use other APIs like JinaChat, LocalAI and Prem.
|
|
"""
|
|
|
|
|
|
INPUT_KEY_INFO = """The variable to be used as Chat Input when more than one variable is available."""
|
|
OUTPUT_KEY_INFO = """The variable to be used as Chat Output (e.g. answer in a ConversationalRetrievalChain)"""
|
|
|
|
|
|
CLASSES_TO_REMOVE = ["Serializable", "BaseModel", "object"]
|