fix: minor fixes

weavel/_worker.py

@@ -40,8 +40,7 @@
 from weavel._buffer_storage import BufferStorage
 from weavel._api_client import APIClient, AsyncAPIClient
 from weavel.utils import logger
-from weavel.types.datasets import DatasetItem, Dataset, Prompt, PromptVersion
-from weavel.types.response_format import ResponseFormat
+from weavel.types import DatasetItem, Dataset, Prompt, PromptVersion, ResponseFormat
 
 class Worker:
     _instance = None

weavel/client.py

@@ -8,7 +8,7 @@
 import time
 from typing import Callable, Dict, List, Optional, Any, Union
 from uuid import uuid4
-from weavel.types.response_format import ResponseFormat
+from weavel.types import ResponseFormat
 
 from dotenv import load_dotenv
 from weavel._worker import Worker

weavel/types/__init__.py

@@ -2,3 +2,4 @@
 from .observations import *
 from .records import *
 from .session import *
+from .response_format import *

weavel/types/datasets.py

@@ -1,6 +1,6 @@
 from typing import Any, Dict, List, Optional, Union
 from pydantic import BaseModel
-from response_format import ResponseFormat
+from weavel.types import ResponseFormat
 
 class DatasetItem(BaseModel):
     uuid: Optional[str] = None