fix linting ci build failures

Author: atulikumwenayo

src/datacustomcode/config.py

@@ -37,12 +37,12 @@
 # This lets all readers and writers to be findable via config
 from datacustomcode.io import *  # noqa: F403
 from datacustomcode.io.base import BaseDataAccessLayer
-from datacustomcode.io.reader.base import BaseDataCloudReader
-from datacustomcode.io.writer.base import BaseDataCloudWriter
 from datacustomcode.spark.base import BaseSparkSessionProvider
 
 if TYPE_CHECKING:
     from pyspark.sql import SparkSession
+    from datacustomcode.io.reader.base import BaseDataCloudReader
+    from datacustomcode.io.writer.base import BaseDataCloudWriter
 
 
 _T = TypeVar("_T", bound="BaseDataAccessLayer")
@@ -82,8 +82,8 @@ def to_object(self) -> _P:
 
 
 class ClientConfig(BaseConfig):
-    reader_config: Union[AccessLayerObjectConfig[BaseDataCloudReader], None] = None
-    writer_config: Union[AccessLayerObjectConfig[BaseDataCloudWriter], None] = None
+    reader_config: Union[AccessLayerObjectConfig["BaseDataCloudReader"], None] = None
+    writer_config: Union[AccessLayerObjectConfig["BaseDataCloudWriter"], None] = None
     spark_config: Union[SparkConfig, None] = None
     spark_provider_config: Union[
         SparkProviderConfig[BaseSparkSessionProvider], None

src/datacustomcode/llm_gateway/default.py

@@ -19,8 +19,8 @@
     Optional,
 )
 
-import requests
 from loguru import logger
+import requests
 
 from datacustomcode.einstein_platform_client import EinsteinPlatformClient
 from datacustomcode.llm_gateway.base import LLMGateway
@@ -44,8 +44,7 @@ def __init__(
 
     def generate_text(self, request: GenerateTextRequest) -> GenerateTextResponse:
         api_url = (
-            f"{self.EINSTEIN_PLATFORM_URL}/models/"
-            f"{request.model_name}/generations"
+            f"{self.EINSTEIN_PLATFORM_URL}/models/{request.model_name}/generations"
         )
 
         payload: Dict[str, Any] = {"prompt": request.prompt}

src/datacustomcode/templates/function/payload/entrypoint.py

@@ -80,8 +80,6 @@ def generate_text(runtime: Runtime):
         print(llm_response.error_code)
 
 
-
-
 def function(request: dict, runtime: Runtime) -> dict:
     logger.info("Inside Function")
     logger.info(request)