diff --git a/docs/docs/Components/bundles-perplexity.mdx b/docs/docs/Components/bundles-perplexity.mdx index fceaac1c3..e9aa6054c 100644 --- a/docs/docs/Components/bundles-perplexity.mdx +++ b/docs/docs/Components/bundles-perplexity.mdx @@ -29,9 +29,8 @@ You can toggle parameters through the LanguageModel: # type: ignore[type-var] api_key = SecretStr(self.api_key).get_secret_value() temperature = self.temperature model = self.model_name - max_output_tokens = self.max_output_tokens - top_k = self.top_k + max_tokens = self.max_tokens top_p = self.top_p n = self.n @@ -78,8 +69,7 @@ class PerplexityComponent(LCModelComponent): model=model, temperature=temperature or 0.75, pplx_api_key=api_key, - top_k=top_k or None, top_p=top_p or None, n=n or 1, - max_output_tokens=max_output_tokens, + max_tokens=max_tokens, )