-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathsettings.py
More file actions
40 lines (33 loc) · 1.39 KB
/
settings.py
File metadata and controls
40 lines (33 loc) · 1.39 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import os
# === Environment variables ===
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY", "")
PINECONE_API_KEY = os.environ.get("PINECONE_API_KEY", "")
PINECONE_INDEX = os.environ.get("PINECONE_INDEX", "lsfusion")
PINECONE_NAMESPACE = os.environ.get("PINECONE_NAMESPACE", "")
# Embedding model
EMBEDDING_MODEL = os.environ.get("EMBEDDING_MODEL", "text-embedding-3-large")
# Metadata keys
TEXT = "text"
SOURCETYPE = "sourceType"
SOURCETYPE_DOCUMENTATION = "documentation"
SOURCETYPE_DOCUMENTATION_PARADIGM = "paradigm"
SOURCETYPE_DOCUMENTATION_HOWTO = "how-to"
SOURCETYPE_DOCUMENTATION_LANGUAGE = "language"
SOURCETYPE_DOCUMENTATION_TUTORIAL = "tutorial"
SOURCETYPE_SAMPLES = "samples"
SOURCETYPE_ARTICLES = "articles"
SOURCETYPE_TALKS = "talks"
# Canonical combined source-type identifiers
SOURCETYPE_DOC_PARADIGM = f"{SOURCETYPE_DOCUMENTATION}-{SOURCETYPE_DOCUMENTATION_PARADIGM}"
SOURCETYPE_DOC_HOWTO = f"{SOURCETYPE_DOCUMENTATION}-{SOURCETYPE_DOCUMENTATION_HOWTO}"
SOURCETYPE_DOC_LANGUAGE = f"{SOURCETYPE_DOCUMENTATION}-{SOURCETYPE_DOCUMENTATION_LANGUAGE}"
SOURCETYPE_DOC_TUTORIAL = f"{SOURCETYPE_DOCUMENTATION}-{SOURCETYPE_DOCUMENTATION_TUTORIAL}"
# TopK per source type (mirrors your Java setup, but without 'system')
TOP_K = {
SOURCETYPE_DOC_PARADIGM: 3,
SOURCETYPE_DOC_HOWTO: 3,
SOURCETYPE_DOC_LANGUAGE: 3,
SOURCETYPE_DOC_TUTORIAL: 3,
SOURCETYPE_ARTICLES: 3,
SOURCETYPE_TALKS: 3,
}