File size: 626 Bytes
e3278e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from typing import List, Optional
from ..types.utils import (
Embedding,
EmbeddingResponse,
ImageObject,
ImageResponse,
Usage,
)
def mock_embedding(model: str, mock_response: Optional[List[float]]):
if mock_response is None:
mock_response = [0.0] * 1536
return EmbeddingResponse(
model=model,
data=[Embedding(embedding=mock_response, index=0, object="embedding")],
usage=Usage(prompt_tokens=10, completion_tokens=0),
)
def mock_image_generation(model: str, mock_response: str):
return ImageResponse(
data=[ImageObject(url=mock_response)],
)
|