Vladimir Blagojevic 5ee393226d

fix: Support all SageMaker HF text generation models (other than Falcon) (#5205) ...

* Create SageMaker base class and two implementation subclasses
---------

Co-authored-by: Stefano Fiorucci <44616784+anakin87@users.noreply.github.com>

2023-06-26 19:59:16 +02:00

..

invocation_layer

fix: Support all SageMaker HF text generation models (other than Falcon) (#5205)

2023-06-26 19:59:16 +02:00

__init__.py

Make promptnode test unit (#4420)

2023-03-15 22:17:23 +01:00

conftest.py

test: Rework conftest (#4614)

2023-04-11 10:33:43 +02:00

test_handlers.py

chore: block all HTTP requests in CI (#5088)

2023-06-13 14:52:24 +02:00

test_prompt_model.py

feat: pass model parameters to HFLocalInvocationLayer via model_kwargs, enabling direct model usage (#4956)

2023-06-07 13:34:45 +02:00

test_prompt_node.py

feat: add ensure token limit for direct prompting of ChatGPT (#5166)

2023-06-21 15:41:28 +02:00

test_prompt_template.py

fix: PromptNode falls back to empty list of documents if none are provided but expected (#5132)

2023-06-13 16:35:19 +02:00