refactor: consider using method overload

[inspiralpatterns]

llmflows/llmflows/llms/openai_embeddings.py

Lines 50 to 52 in 6ce79a8

	def generate(
	self, docs: Union[VectorDoc, list[VectorDoc]]
	) -> Union[VectorDoc, list[VectorDoc]]:

I am not a fan of functions that can return either one or the other type, however, I assume that given a List[VectorDoc], generate will always return a List[VectorDoc]. If so, then you can use the @overload decorator and declare two function signatures:

from typing import overload

@overload
def generate(docs: VectorDoc) -> VectorDoc:
    ...

@overload
def generate(docs: List[VectorDoc]) -> List[VectorDoc]:
    ...

def generate(x):
    # here goes the implementation

Note that the ellipsis ... is required.