[TLM] Add custom types to docstrings

[ulya-tkch]

This change will allow our custom defined types to show up on help.cleanlab.ai here

[jwmueller]

@ulya-tkch I unfortunately don't find the new descriptions super clear either. Perhaps you could first explain everything as if it were just floats or list of floats. And then later in the docstring say:

Actually instead of float, this may be a TLMScore object if ... (important to specify when the returned object might not be a float / the simpler return type, and what the type of the returned object would be in those scenarios).

Will leave reviews of this to @jas2600

But encourage you to write as simply and clearly as you can.

[ulya-tkch]

MAX_PROMPT_LENGTH_TOKENS needs to be increased for TLM unit tests to start passing again.

[jas2600]

two small suggested changes, otherwise LGTM