klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-06 13:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pass `model' arg to extract_search_type method from beta search API

Browse Source 
Issue caught by mypy
This commit is contained in:
Debanjum Singh Solanky
2023-01-09 16:59:45 -03:00
parent 8498903641
commit 8dc6ee8b6c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/routers/api_beta.py
View File

@@ -22,9 +22,13 @@ logger = logging.getLogger(__name__)
# Create Routes # Create Routes
@api_beta.get('/search') @api_beta.get('/search')
def search_beta(q: str, n: Optional[int] = 1): def search_beta(q: str, n: Optional[int] = 1):
# Initialize Variables
model = state.processor_config.conversation.model
api_key = state.processor_config.conversation.openai_api_key
# Extract Search Type using GPT # Extract Search Type using GPT
try: try:
metadata = extract_search_type(q, api_key=state.processor_config.conversation.openai_api_key, verbose=state.verbose) metadata = extract_search_type(q, model=model, api_key=api_key, verbose=state.verbose)
search_type = get_from_dict(metadata, "search-type") search_type = get_from_dict(metadata, "search-type")
except Exception as e: except Exception as e:
return {'status': 'error', 'result': [str(e)], 'type': None} return {'status': 'error', 'result': [str(e)], 'type': None}