Spaces:
Sleeping
Sleeping
WilliamGazeley
commited on
Commit
·
ef4505e
1
Parent(s):
c34800f
Bump top k to 3
Browse files- src/config.py +2 -2
- src/functions.py +1 -1
src/config.py
CHANGED
@@ -9,8 +9,8 @@ class Config(BaseSettings):
|
|
9 |
az_search_endpoint: str = Field("https://analysis-bank.search.windows.net")
|
10 |
az_search_api_key: str = Field(...)
|
11 |
az_search_idx_name: str = Field("analysis-index")
|
12 |
-
az_search_top_k: int = Field(
|
13 |
-
az_search_min_score: float = Field(
|
14 |
|
15 |
chat_template: str = Field("chatml", description="Chat template for prompt formatting")
|
16 |
num_fewshot: int | None = Field(None, description="Option to use json mode examples")
|
|
|
9 |
az_search_endpoint: str = Field("https://analysis-bank.search.windows.net")
|
10 |
az_search_api_key: str = Field(...)
|
11 |
az_search_idx_name: str = Field("analysis-index")
|
12 |
+
az_search_top_k: int = Field(3, description="Max number of results to retrun")
|
13 |
+
az_search_min_score: float = Field(9.0, description="Only results above this confidence score is used")
|
14 |
|
15 |
chat_template: str = Field("chatml", description="Chat template for prompt formatting")
|
16 |
num_fewshot: int | None = Field(None, description="Option to use json mode examples")
|
src/functions.py
CHANGED
@@ -22,7 +22,7 @@ az_search_client = SearchClient(config.az_search_endpoint, config.az_search_idx_
|
|
22 |
@tool
|
23 |
def get_analysis(query: str) -> dict:
|
24 |
"""
|
25 |
-
Searches through your database of company and crypto analysis, retrieves top
|
26 |
pieces of analysis relevant to your query. You MUST ALWAYS use this function
|
27 |
to help form your opinions and predictions.
|
28 |
|
|
|
22 |
@tool
|
23 |
def get_analysis(query: str) -> dict:
|
24 |
"""
|
25 |
+
Searches through your database of company and crypto analysis, retrieves top 3
|
26 |
pieces of analysis relevant to your query. You MUST ALWAYS use this function
|
27 |
to help form your opinions and predictions.
|
28 |
|