Merge pull request deepgram#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):