Fix Google Colab Demo (onyx-dot-app#275)

itsnotvalid · Aug 9, 2023 · a03818e · a03818e
1 parent 89f71ac
commit a03818e
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/backend/danswer/direct_qa/request_model.py b/backend/danswer/direct_qa/request_model.py
@@ -151,7 +151,7 @@ def send_model_request(
         }
 
         data = {
-            "input": filled_prompt,
+            "inputs": filled_prompt,
             "parameters": {
                 "temperature": 0.0,
                 "max_tokens": max_output_tokens,
@@ -212,6 +212,10 @@ def __init__(
         self.model_class = get_host_specific_model_class(model_host_type)
         self.timeout = timeout
 
+    @property
+    def requires_api_key(self) -> bool:
+        return self.model_class.requires_api_key
+
     def _get_request_response(
         self, query: str, context_docs: list[InferenceChunk], stream: bool
     ) -> Response: