adjust order to be after init

run-llama · Sep 6, 2024 · 667513f · 667513f
1 parent d9c04b8
commit 667513f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/llama-index-integrations/llms/llama-index-llms-reka/llama_index/llms/reka/base.py b/llama-index-integrations/llms/llama-index-llms-reka/llama_index/llms/reka/base.py
@@ -119,16 +119,15 @@ def __init__(
  "Reka API key is required. Please provide it as an argument or set the REKA_API_KEY environment variable."
  )
 
- self._client = Reka(api_key=api_key)
- self._aclient = AsyncReka(api_key=api_key)
-
  super().__init__(
  model=model,
  temperature=temperature,
  max_tokens=max_tokens,
  additional_kwargs=additional_kwargs,
  callback_manager=callback_manager,
  )
+ self._client = Reka(api_key=api_key)
+ self._aclient = AsyncReka(api_key=api_key)
 
  @property
  def metadata(self) -> LLMMetadata:

diff --git a/...ti_modal_llms/llama-index-multi-modal-llms-reka/llama_index/multi_modal_llms/reka/base.py b/...ti_modal_llms/llama-index-multi-modal-llms-reka/llama_index/multi_modal_llms/reka/base.py
@@ -99,9 +99,6 @@ def __init__(
  "Reka API key is required. Please provide it as an argument or set the REKA_API_KEY environment variable."
  )
 
- self._client = Reka(api_key=api_key)
- self._aclient = AsyncReka(api_key=api_key)
-
  super().__init__(
  model=model,
  temperature=temperature,
@@ -110,6 +107,9 @@ def __init__(
  callback_manager=callback_manager,
  )
 
+ self._client = Reka(api_key=api_key)
+ self._aclient = AsyncReka(api_key=api_key)
+
  @classmethod
  def class_name(cls) -> str:
  return "Reka_MultiModal_LLM"