community[patch]: Add missing annotations (#24890)

This PR adds annotations in comunity package.

Annotations are only strictly needed in subclasses of BaseModel for
pydantic 2 compatibility.

This PR adds some unnecessary annotations, but they're not bad to have
regardless for documentation pages.
This commit is contained in:
Eugene Yurtsev
2024-07-31 14:13:44 -04:00
committed by GitHub
parent 7720483432
commit d24b82357f
44 changed files with 98 additions and 91 deletions

View File

@@ -59,8 +59,8 @@ def test_custom_formatter() -> None:
"""Test ability to create a custom content formatter."""
class CustomFormatter(ContentFormatterBase):
content_type = "application/json"
accepts = "application/json"
content_type: str = "application/json"
accepts: str = "application/json"
def format_request_payload(self, prompt: str, model_kwargs: Dict) -> bytes: # type: ignore[override]
input_str = json.dumps(
@@ -101,8 +101,8 @@ def test_invalid_request_format() -> None:
"""Test invalid request format."""
class CustomContentFormatter(ContentFormatterBase):
content_type = "application/json"
accepts = "application/json"
content_type: str = "application/json"
accepts: str = "application/json"
def format_request_payload(self, prompt: str, model_kwargs: Dict) -> bytes: # type: ignore[override]
input_str = json.dumps(