Merge pull request #537 from deepgram/feat/context

feat: support context length option

Author: naomi-lgbt

deepgram/clients/agent/v1/websocket/options.py

@@ -2,7 +2,7 @@
 # Use of this source code is governed by a MIT license that can be found in the LICENSE file.
 # SPDX-License-Identifier: MIT
 
-from typing import List, Optional, Union, Any, Tuple, Dict
+from typing import List, Optional, Union, Any, Tuple, Dict, Literal
 import logging
 
 from dataclasses import dataclass, field
@@ -150,6 +150,9 @@ class Think(BaseResponse):
     prompt: Optional[str] = field(
         default=None, metadata=dataclass_config(exclude=lambda f: f is None)
     )
+    context_length: Optional[Union[int, Literal["max"]]] = field(
+        default=None, metadata=dataclass_config(exclude=lambda f: f is None)
+    )
 
     def __post_init__(self):
         if not isinstance(self.provider, Provider):