WilliamGazeley commited on
Commit
ef4505e
1 Parent(s): c34800f

Bump top k to 3

Browse files
Files changed (2) hide show
  1. src/config.py +2 -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(2, description="Max number of results to retrun")
13
- az_search_min_score: float = Field(12.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")
 
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 2
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