fix: unify types to work on android

Author: mateuszlampert

packages/react-native-executorch/common/rnexecutorch/models/llm/LLM.cpp

@@ -86,7 +86,7 @@ size_t LLM::getPromptTokenCount() const noexcept {
   return runner->stats_.num_prompt_tokens;
 }
 
-size_t LLM::countTextTokens(std::string text) const {
+int32_t LLM::countTextTokens(std::string text) const {
   if (!runner || !runner->is_loaded()) {
     throw RnExecutorchError(
         RnExecutorchErrorCode::ModuleNotLoaded,
@@ -147,7 +147,7 @@ void LLM::setTopp(float topp) {
   runner->set_topp(topp);
 }
 
-size_t LLM::getMaxContextLength() const {
+int32_t LLM::getMaxContextLength() const {
   if (!runner || !runner->is_loaded()) {
     throw RnExecutorchError(
         RnExecutorchErrorCode::ModuleNotLoaded,

packages/react-native-executorch/common/rnexecutorch/models/llm/LLM.h

@@ -25,13 +25,13 @@ class LLM : public BaseModel {
   void unload() noexcept;
   size_t getGeneratedTokenCount() const noexcept;
   size_t getPromptTokenCount() const noexcept;
-  size_t countTextTokens(std::string text) const;
+  int32_t countTextTokens(std::string text) const;
   size_t getMemoryLowerBound() const noexcept;
   void setCountInterval(size_t countInterval);
   void setTemperature(float temperature);
   void setTopp(float topp);
   void setTimeInterval(size_t timeInterval);
-  size_t getMaxContextLength() const;
+  int32_t getMaxContextLength() const;
 
 private:
   std::unique_ptr<example::Runner> runner;

packages/react-native-executorch/common/runner/runner.cpp

@@ -342,14 +342,14 @@ void Runner::set_topp(float topp) noexcept {
   }
 }
 
-size_t Runner::get_max_context_length() const {
+int32_t Runner::get_max_context_length() const {
   if (!is_loaded()) {
     return metadata_.at(kMaxContextLen);
   }
   return config_.max_context_length;
 }
 
-size_t Runner::count_text_tokens(const std::string &text) const {
+int32_t Runner::count_text_tokens(const std::string &text) const {
   auto encodeResult =
       tokenizer_->encode(text, numOfAddedBoSTokens, numOfAddedEoSTokens);
 

packages/react-native-executorch/common/runner/runner.h

@@ -50,8 +50,8 @@ class Runner : public llm::IRunner {
   void set_time_interval(size_t time_interval);
   void set_temperature(float temperature) noexcept;
   void set_topp(float topp) noexcept;
-  size_t count_text_tokens(const std::string &text) const;
-  size_t get_max_context_length() const;
+  int32_t count_text_tokens(const std::string &text) const;
+  int32_t get_max_context_length() const;
 
   void stop() override;
   void reset() override;