From 8dc6ee8b6c66ee75a33a6beb98e55c4674e0284d Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Mon, 9 Jan 2023 16:59:45 -0300 Subject: [PATCH] Pass `model' arg to extract_search_type method from beta search API Issue caught by mypy --- src/routers/api_beta.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/routers/api_beta.py b/src/routers/api_beta.py index 552a5945..a92443db 100644 --- a/src/routers/api_beta.py +++ b/src/routers/api_beta.py @@ -22,9 +22,13 @@ logger = logging.getLogger(__name__) # Create Routes @api_beta.get('/search') 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 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") except Exception as e: return {'status': 'error', 'result': [str(e)], 'type': None}