Skip to content

Commit

Permalink
Merge to prod (#323)
Browse files Browse the repository at this point in the history
* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Merge dev to test (#107)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Development to Test (#109)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

---------

Co-authored-by: Lucas Pavanelli <[email protected]>

* Merge to test (#111)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

---------

Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update Finetuner functional tests (#112)

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#118)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#117)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Do not download textual URLs (#120)

* Do not download textual URLs

* Treat as string

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enable api key parameter in data asset creation (#122)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Merge to test (#124)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#117)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Do not download textual URLs (#120)

* Do not download textual URLs

* Treat as string

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enable api key parameter in data asset creation (#122)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: mikelam-us-aixplain <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update Finetuner hyperparameters (#125)

* Update Finetuner hyperparameters

* Change hyperparameters error message

* Merge dev to test (#126)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#117)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Do not download textual URLs (#120)

* Do not download textual URLs

* Treat as string

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enable api key parameter in data asset creation (#122)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update Finetuner hyperparameters (#125)

* Update Finetuner hyperparameters

* Change hyperparameters error message

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: mikelam-us-aixplain <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Add new LLMs finetuner models (mistral and solar) (#128)

* Merge dev to test (#129)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#117)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Do not download textual URLs (#120)

* Do not download textual URLs

* Treat as string

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enable api key parameter in data asset creation (#122)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update Finetuner hyperparameters (#125)

* Update Finetuner hyperparameters

* Change hyperparameters error message

* Add new LLMs finetuner models (mistral and solar) (#128)

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: mikelam-us-aixplain <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enabling dataset ID and model ID as parameters for finetuner creation (#131)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Fix supplier representation of a model (#132)

* Fix supplier representation of a model

* Fixing parameter typing

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Fixing indentation in documentation sample code (#134)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Merge to test (#135)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#117)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Do not download textual URLs (#120)

* Do not download textual URLs

* Treat as string

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enable api key parameter in data asset creation (#122)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update Finetuner hyperparameters (#125)

* Update Finetuner hyperparameters

* Change hyperparameters error message

* Add new LLMs finetuner models (mistral and solar) (#128)

* Enabling dataset ID and model ID as parameters for finetuner creation (#131)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Fix supplier representation of a model (#132)

* Fix supplier representation of a model

* Fixing parameter typing

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Fixing indentation in documentation sample code (#134)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: mikelam-us-aixplain <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update FineTune unit and functional tests (#136)

* Merge dev to test (#137)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#117)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Do not download textual URLs (#120)

* Do not download textual URLs

* Treat as string

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Enable api key parameter in data asset creation (#122)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update Finetuner hyperparameters (#125)

* Update Finetuner hyperparameters

* Change hyperparameters error message

* Add new LLMs finetuner models (mistral and solar) (#128)

* Enabling dataset ID and model ID as parameters for finetuner creation (#131)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Fix supplier representation of a model (#132)

* Fix supplier representation of a model

* Fixing parameter typing

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Fixing indentation in documentation sample code (#134)

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Update FineTune unit and functional tests (#136)

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: mikelam-us-aixplain <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Click fix (#140)

* Merge to prod (#119)

* Merge dev to test (#107)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Development to Test (#109)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

---------

Co-authored-by: Lucas Pavanelli <[email protected]>

* Merge to test (#111)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

---------

Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Merge dev to test (#113)

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Update Finetuner functional tests (#112)

---------

Co-authored-by: Thiago Castro Ferreira <[email protected]>

* Hf deployment test (#114)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Hf deployment test (#118)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Create bounds for FineTune hyperparameters (#103)

* Test bound to hyperparameters

* Update finetune llm hyperparameters

* Remove option to use PEFT, always on use now

* Fixing pipeline general asset test (#106)

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Update Finetuner functional tests (#112)

* Hf deployment test (#115)

* Started adding Hugging Face deployment to aiXplain SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added model status function to  SDK

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updating

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Updated CLI

Signed-off-by: mikelam-us <[email protected]>

* Adding CLI

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Corrected request error

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Clearing out unnecessary information in return

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Simplifying status

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests and correcting tokens

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added bad repo ID test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Finished rough draft of tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding tests

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Fixing hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding hf token

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Correcting first test

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Testing

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Adding config

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added user doc

Signed-off-by: mikelam-us-aixplain <[email protected]>

* Added gated model test

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>

* Adding HF token

Signed-off-by: mikelam-us-aixplain <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>

---------

Signed-off-by: mikelam-us-aixplain <[email protected]>
Signed-off-by: mikelam-us <[email protected]>
Co-authored-by: Lucas Pavanelli <[email protected]>
Co-authored-by: Thiago Castro Ferreira <[email protected]>
Co-authored-by: mik…
  • Loading branch information
19 people authored Nov 26, 2024
1 parent 15416d4 commit 944df99
Show file tree
Hide file tree
Showing 34 changed files with 307 additions and 207 deletions.
2 changes: 1 addition & 1 deletion aixplain/decorators/api_key_checker.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

def check_api_key(method):
def wrapper(*args, **kwargs):
if config.TEAM_API_KEY == "":
if config.TEAM_API_KEY == "" and config.AIXPLAIN_API_KEY == "":
raise Exception(
"A 'TEAM_API_KEY' is required to run an asset. For help, please refer to the documentation (https://github.com/aixplain/aixplain#api-key-setup)"
)
Expand Down
9 changes: 2 additions & 7 deletions aixplain/enums/function.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,15 +31,11 @@

def load_functions():
api_key = config.TEAM_API_KEY
aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

url = urljoin(backend_url, "sdk/functions")
if aixplain_key != "":
api_key = aixplain_key
headers = {"x-aixplain-key": aixplain_key, "Content-Type": "application/json"}
else:
headers = {"x-api-key": api_key, "Content-Type": "application/json"}

headers = {"x-api-key": api_key, "Content-Type": "application/json"}
r = _request_with_retry("get", url, headers=headers)
if not 200 <= r.status_code < 300:
raise Exception(
Expand All @@ -61,5 +57,4 @@ def load_functions():
}
return functions, functions_input_output


Function, FunctionInputOutput = load_functions()
8 changes: 2 additions & 6 deletions aixplain/enums/language.py
Original file line number Diff line number Diff line change
Expand Up @@ -31,15 +31,11 @@

def load_languages():
api_key = config.TEAM_API_KEY
aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

url = urljoin(backend_url, "sdk/languages")
if aixplain_key != "":
api_key = aixplain_key
headers = {"x-aixplain-key": aixplain_key, "Content-Type": "application/json"}
else:
headers = {"x-api-key": api_key, "Content-Type": "application/json"}

headers = {"x-api-key": api_key, "Content-Type": "application/json"}
r = _request_with_retry("get", url, headers=headers)
if not 200 <= r.status_code < 300:
raise Exception(
Expand Down
8 changes: 2 additions & 6 deletions aixplain/enums/license.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,15 +32,11 @@
def load_licenses():
try:
api_key = config.TEAM_API_KEY
aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

url = urljoin(backend_url, "sdk/licenses")
if aixplain_key != "":
api_key = aixplain_key
headers = {"x-aixplain-key": aixplain_key, "Content-Type": "application/json"}
else:
headers = {"x-api-key": api_key, "Content-Type": "application/json"}

headers = {"x-api-key": api_key, "Content-Type": "application/json"}
r = _request_with_retry("get", url, headers=headers)
if not 200 <= r.status_code < 300:
raise Exception(
Expand Down
8 changes: 2 additions & 6 deletions aixplain/enums/supplier.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,15 +39,11 @@ def clean_name(name):

def load_suppliers():
api_key = config.TEAM_API_KEY
aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

url = urljoin(backend_url, "sdk/suppliers")
if aixplain_key != "":
api_key = aixplain_key
headers = {"x-aixplain-key": aixplain_key, "Content-Type": "application/json"}
else:
headers = {"x-api-key": api_key, "Content-Type": "application/json"}

headers = {"x-api-key": api_key, "Content-Type": "application/json"}
logging.debug(f"Start service for GET API Creation - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
if not 200 <= r.status_code < 300:
Expand Down
8 changes: 3 additions & 5 deletions aixplain/factories/agent_factory/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,11 +176,9 @@ def get(cls, agent_id: Text, api_key: Optional[Text] = None) -> Agent:
from aixplain.factories.agent_factory.utils import build_agent

url = urljoin(config.BACKEND_URL, f"sdk/agents/{agent_id}")
if config.AIXPLAIN_API_KEY != "":
headers = {"x-aixplain-key": f"{config.AIXPLAIN_API_KEY}", "Content-Type": "application/json"}
else:
api_key = api_key if api_key is not None else config.TEAM_API_KEY
headers = {"x-api-key": api_key, "Content-Type": "application/json"}

api_key = api_key if api_key is not None else config.TEAM_API_KEY
headers = {"x-api-key": api_key, "Content-Type": "application/json"}
logging.info(f"Start service for GET Agent - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down
2 changes: 1 addition & 1 deletion aixplain/factories/asset_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@


class AssetFactory:
aixplain_key = config.AIXPLAIN_API_KEY

backend_url = config.BACKEND_URL

@abstractmethod
Expand Down
23 changes: 6 additions & 17 deletions aixplain/factories/benchmark_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ class BenchmarkFactory:
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY

backend_url = config.BACKEND_URL

@classmethod
Expand All @@ -69,10 +69,8 @@ def _get_benchmark_jobs_from_benchmark_id(cls, benchmark_id: Text) -> List[Bench
List[BenchmarkJob]: List of associated benchmark jobs
"""
url = urljoin(cls.backend_url, f"sdk/benchmarks/{benchmark_id}/jobs")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
job_list = [cls._create_benchmark_job_from_response(job_info) for job_info in resp]
Expand Down Expand Up @@ -107,10 +105,7 @@ def get(cls, benchmark_id: str) -> Benchmark:
resp = None
try:
url = urljoin(cls.backend_url, f"sdk/benchmarks/{benchmark_id}")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
logging.info(f"Start service for GET Benchmark - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down Expand Up @@ -145,10 +140,7 @@ def get_job(cls, job_id: Text) -> BenchmarkJob:
BenchmarkJob: Created 'BenchmarkJob' object
"""
url = urljoin(cls.backend_url, f"sdk/benchmarks/jobs/{job_id}")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
benchmarkJob = cls._create_benchmark_job_from_response(resp)
Expand Down Expand Up @@ -235,10 +227,7 @@ def list_normalization_options(cls, metric: Metric, model: Model) -> List[str]:
"""
try:
url = urljoin(cls.backend_url, "sdk/benchmarks/normalization-options")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
payload = json.dumps({"metricId": metric.id, "modelIds": [model.id]})
r = _request_with_retry("post", url, headers=headers, data=payload)
resp = r.json()
Expand Down
13 changes: 4 additions & 9 deletions aixplain/factories/corpus_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,6 @@


class CorpusFactory(AssetFactory):
aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

@classmethod
Expand Down Expand Up @@ -117,10 +116,8 @@ def get(cls, corpus_id: Text) -> Corpus:
"""
try:
url = urljoin(cls.backend_url, f"sdk/corpora/{corpus_id}/overview")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
logging.info(f"Start service for GET Corpus - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down Expand Up @@ -172,10 +169,8 @@ def list(
Dict: list of corpora in agreement with the filters, page number, page total and total elements
"""
url = urljoin(cls.backend_url, "sdk/corpora/paginate")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

assert 0 < page_size <= 100, "Corpus List Error: Page size must be greater than 0 and not exceed 100."
payload = {"pageSize": page_size, "pageNumber": page_number, "sort": [{"field": "createdAt", "dir": -1}]}
Expand Down
7 changes: 2 additions & 5 deletions aixplain/factories/data_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,6 @@ class DataFactory(AssetFactory):
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

@classmethod
Expand Down Expand Up @@ -92,10 +91,8 @@ def get(cls, data_id: Text) -> Data:
Data: Created 'Data' object
"""
url = urljoin(cls.backend_url, f"sdk/data/{data_id}/overview")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
logging.info(f"Start service for GET Data - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down
14 changes: 5 additions & 9 deletions aixplain/factories/dataset_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ class DatasetFactory(AssetFactory):
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY

backend_url = config.BACKEND_URL

@classmethod
Expand Down Expand Up @@ -164,10 +164,8 @@ def get(cls, dataset_id: Text) -> Dataset:
"""
try:
url = urljoin(cls.backend_url, f"sdk/datasets/{dataset_id}/overview")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
logging.info(f"Start service for GET Dataset - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down Expand Up @@ -214,10 +212,8 @@ def list(
Dict: list of datasets in agreement with the filters, page number, page total and total elements
"""
url = urljoin(cls.backend_url, "sdk/datasets/paginate")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

assert 0 < page_size <= 100, "Dataset List Error: Page size must be greater than 0 and not exceed 100."
payload = {
Expand Down
1 change: 0 additions & 1 deletion aixplain/factories/finetune_factory/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,6 @@ class FinetuneFactory:
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

@classmethod
Expand Down
11 changes: 2 additions & 9 deletions aixplain/factories/metric_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,6 @@ class MetricFactory:
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

@classmethod
Expand Down Expand Up @@ -76,10 +75,7 @@ def get(cls, metric_id: Text) -> Metric:
resp, status_code = None, 200
try:
url = urljoin(cls.backend_url, f"sdk/metrics/{metric_id}")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
logging.info(f"Start service for GET Metric - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down Expand Up @@ -126,10 +122,7 @@ def list(
if is_reference_required is not None:
filter_params["referenceRequired"] = 1 if is_reference_required else 0

if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
r = _request_with_retry("get", url, headers=headers, params=filter_params)
resp = r.json()
logging.info(f"Listing Metrics: Status of getting metrics: {resp}")
Expand Down
13 changes: 4 additions & 9 deletions aixplain/factories/model_factory.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,6 @@ class ModelFactory:
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

@classmethod
Expand Down Expand Up @@ -107,10 +106,8 @@ def get(cls, model_id: Text, api_key: Optional[Text] = None) -> Model:
resp = None
try:
url = urljoin(cls.backend_url, f"sdk/models/{model_id}")
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}
logging.info(f"Start service for GET Model - {url} - {headers}")
r = _request_with_retry("get", url, headers=headers)
resp = r.json()
Expand Down Expand Up @@ -196,10 +193,8 @@ def _get_assets_from_page(
filter_params["sort"] = [{"dir": sort_order.value, "field": sort_by.value}]
if len(lang_filter_params) != 0:
filter_params["ioFilter"] = lang_filter_params
if cls.aixplain_key != "":
headers = {"x-aixplain-key": f"{cls.aixplain_key}", "Content-Type": "application/json"}
else:
headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

headers = {"Authorization": f"Token {config.TEAM_API_KEY}", "Content-Type": "application/json"}

logging.info(f"Start service for POST Models Paginate - {url} - {headers} - {json.dumps(filter_params)}")
r = _request_with_retry("post", url, headers=headers, json=filter_params)
Expand Down
24 changes: 4 additions & 20 deletions aixplain/factories/pipeline_factory/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@ class PipelineFactory:
backend_url (str): The URL for the backend.
"""

aixplain_key = config.AIXPLAIN_API_KEY
backend_url = config.BACKEND_URL

@classmethod
Expand All @@ -65,11 +64,6 @@ def get(cls, pipeline_id: Text, api_key: Optional[Text] = None) -> Pipeline:
"Authorization": f"Token {api_key}",
"Content-Type": "application/json",
}
elif cls.aixplain_key != "":
headers = {
"x-aixplain-key": f"{cls.aixplain_key}",
"Content-Type": "application/json",
}
else:
headers = {
"Authorization": f"Token {config.TEAM_API_KEY}",
Expand Down Expand Up @@ -125,13 +119,8 @@ def get_assets_from_page(cls, page_number: int) -> List[Pipeline]:
"""
try:
url = urljoin(cls.backend_url, f"sdk/pipelines/?pageNumber={page_number}")
if cls.aixplain_key != "":
headers = {
"x-aixplain-key": f"{cls.aixplain_key}",
"Content-Type": "application/json",
}
else:
headers = {

headers = {
"Authorization": f"Token {config.TEAM_API_KEY}",
"Content-Type": "application/json",
}
Expand Down Expand Up @@ -181,13 +170,8 @@ def list(
) -> Dict:

url = urljoin(cls.backend_url, "sdk/pipelines/paginate")
if cls.aixplain_key != "":
headers = {
"x-aixplain-key": f"{cls.aixplain_key}",
"Content-Type": "application/json",
}
else:
headers = {

headers = {
"Authorization": f"Token {config.TEAM_API_KEY}",
"Content-Type": "application/json",
}
Expand Down
7 changes: 2 additions & 5 deletions aixplain/factories/team_agent_factory/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,11 +153,8 @@ def list(cls) -> Dict:
def get(cls, agent_id: Text, api_key: Optional[Text] = None) -> TeamAgent:
"""Get agent by id."""
url = urljoin(config.BACKEND_URL, f"sdk/agent-communities/{agent_id}")
if config.AIXPLAIN_API_KEY != "":
headers = {"x-aixplain-key": f"{config.AIXPLAIN_API_KEY}", "Content-Type": "application/json"}
else:
api_key = api_key if api_key is not None else config.TEAM_API_KEY
headers = {"x-api-key": api_key, "Content-Type": "application/json"}
api_key = api_key if api_key is not None else config.TEAM_API_KEY
headers = {"x-api-key": api_key, "Content-Type": "application/json"}
logging.info(f"Start service for GET Team Agent - {url} - {headers}")
try:
r = _request_with_retry("get", url, headers=headers)
Expand Down
Loading

0 comments on commit 944df99

Please sign in to comment.