Merging from main

Author: ritaagarwala-sf

CHANGELOG.md

@@ -1,5 +1,28 @@
 # Changelog
 
+## 3.0.0
+
+### Breaking Changes
+
+- **Added  `runtime: datacustomcode.runtime.function.Runtime` to function contract for codeType `function`.
+
+  Function now mandates runtime as arguments.
+
+  **Why:** `runTime` allows access to resources ( llm_gateway / file ) available during function execution.
+
+  **Migration:** use function(request: dict, runTime: Runtime) instead od function(request: dict)
+
+  ```python
+  # Before
+  def function(request: dict):
+    pass
+
+  # After
+  def function(request: dict, runTime: Runtime):
+    pass
+  ```
+
+
 ## 2.0.0
 
 ### Breaking Changes

src/datacustomcode/io/reader/query_api.py

@@ -197,7 +197,7 @@ def read_dlo(
             self, "_sf_cli_reader", None
         )
         if sf_cli_reader is not None:
-            return sf_cli_reader.read_dlo(name, schema) # type: ignore[no-any-return]
+            return sf_cli_reader.read_dlo(name, schema)  # type: ignore[no-any-return]
 
         query = self._build_query(name)
 
@@ -230,7 +230,7 @@ def read_dmo(
             self, "_sf_cli_reader", None
         )
         if sf_cli_reader is not None:
-            return sf_cli_reader.read_dmo(name, schema) # type: ignore[no-any-return]
+            return sf_cli_reader.read_dmo(name, schema)  # type: ignore[no-any-return]
 
         query = self._build_query(name)
 

tests/test_llm_gateway.py

@@ -1,14 +1,18 @@
 from __future__ import annotations
 
-import pytest
 from pydantic import ValidationError
+import pytest
 
 from datacustomcode.llm_gateway.base import LLMGateway
 from datacustomcode.llm_gateway.default import DefaultLLMGateway
 from datacustomcode.llm_gateway.types.generate_text_request import GenerateTextRequest
-from datacustomcode.llm_gateway.types.generate_text_request_builder import GenerateTextRequestBuilder
+from datacustomcode.llm_gateway.types.generate_text_request_builder import (
+    GenerateTextRequestBuilder,
+)
 from datacustomcode.llm_gateway.types.generate_text_response import GenerateTextResponse
-from datacustomcode.llm_gateway.types.generate_text_response_builder import GenerateTextResponseBuilder
+from datacustomcode.llm_gateway.types.generate_text_response_builder import (
+    GenerateTextResponseBuilder,
+)
 
 
 class TestGenerateTextRequest:
@@ -86,19 +90,16 @@ def test_builder_with_locale_string(self):
         # Verify the localization structure
         assert request.localization is not None
         assert request.localization["defaultLocale"] == "en-US"
-        assert request.localization["inputLocales"] == [{"locale": "en-US", "probability": 1.0}]
+        assert request.localization["inputLocales"] == [
+            {"locale": "en-US", "probability": 1.0}
+        ]
         assert request.localization["expectedLocales"] == ["en-US"]
 
     def test_builder_with_tags(self):
         """Test builder with tags."""
         builder = GenerateTextRequestBuilder()
         tags = {"user": "test", "session": "123"}
-        request = (
-            builder.set_prompt("Hello")
-            .set_model("gpt-4")
-            .set_tags(tags)
-            .build()
-        )
+        request = builder.set_prompt("Hello").set_model("gpt-4").set_tags(tags).build()
         assert request.tags == tags
 
     def test_builder_validates_on_build(self):
@@ -139,8 +140,7 @@ def test_is_success_false_for_non_200(self):
     def test_text_property_success(self):
         """Test text property extracts generated text on success."""
         response = GenerateTextResponse(
-            status_code=200,
-            data={"generation": {"generatedText": "Hello world"}}
+            status_code=200, data={"generation": {"generatedText": "Hello world"}}
         )
         assert response.text == "Hello world"
 
@@ -162,15 +162,14 @@ def test_text_property_handles_missing_nested_fields(self):
     def test_error_code_property(self):
         """Test error_code property extracts error code on error."""
         response = GenerateTextResponse(
-            status_code=400,
-            data={"errorCode": "INVALID_REQUEST"}
+            status_code=400, data={"errorCode": "INVALID_REQUEST"}
         )
         assert response.error_code == "INVALID_REQUEST"
 
     def test_error_code_falls_back_to_status_code(self):
         """Test error_code falls back to status_code if no errorCode in data."""
         response = GenerateTextResponse(status_code=500, data={"message": "error"})
-        assert response.error_code == '500'
+        assert response.error_code == "500"
 
     def test_error_code_returns_empty_on_success(self):
         """Test error_code returns empty string on success."""
@@ -192,7 +191,7 @@ def test_builder_build_from_dict(self):
         response_dict = {
             "version": "v1",
             "status_code": 200,
-            "data": {"generation": {"generatedText": "Hello world"}}
+            "data": {"generation": {"generatedText": "Hello world"}},
         }
         response = GenerateTextResponseBuilder.build(response_dict)
         assert isinstance(response, GenerateTextResponse)
@@ -235,4 +234,4 @@ def test_generate_text_success_response(self):
         response = gateway.generate_text(request)
         assert response.is_success is True
         assert response.status_code == 200
-        assert len(response.text) > 0
+        assert len(response.text) > 0

tests/test_runtime.py

@@ -1,12 +1,12 @@
 from __future__ import annotations
 
 import threading
+
 import pytest
 
-from datacustomcode.runtime.base import BaseRuntime
-from datacustomcode.runtime.function import Runtime
-from datacustomcode.llm_gateway.default import DefaultLLMGateway
 from datacustomcode.file.path.default import DefaultFindFilePath
+from datacustomcode.llm_gateway.default import DefaultLLMGateway
+from datacustomcode.runtime.function import Runtime
 
 
 class TestRuntimeSingleton:
@@ -70,12 +70,12 @@ def setup_method(self):
 
     def test_runtime_has_llm_gateway(self):
         """Test Runtime has llm_gateway property."""
-        assert hasattr(self.runtime, 'llm_gateway')
+        assert hasattr(self.runtime, "llm_gateway")
         assert isinstance(self.runtime.llm_gateway, DefaultLLMGateway)
 
     def test_runtime_has_file(self):
         """Test Runtime has file property."""
-        assert hasattr(self.runtime, 'file')
+        assert hasattr(self.runtime, "file")
         assert isinstance(self.runtime.file, DefaultFindFilePath)
 
     def test_runtime_initializes_only_once(self):
@@ -88,4 +88,4 @@ def test_runtime_initializes_only_once(self):
 
         # Should still be the same instances
         llm_gateway_after = self.runtime.llm_gateway
-        assert llm_gateway_before is llm_gateway_after
+        assert llm_gateway_before is llm_gateway_after