Add disable_structured_output to bedrock examples

Giskard-AI · Nov 21, 2024 · a61e4b2 · a61e4b2
1 parent 82712c7
commit a61e4b2
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/docs/open_source/scan/scan_llm/index.md b/docs/open_source/scan/scan_llm/index.md
@@ -111,7 +111,7 @@ import giskard
 api_base = "http://localhost:11434" # default api_base for local Ollama
 
 # See supported models here: https://docs.litellm.ai/docs/providers/ollama#ollama-models
-giskard.llm.set_llm_model("ollama/llama3", disable_structured_output=True, api_base=api_base)
+giskard.llm.set_llm_model("ollama/llama3.1", disable_structured_output=True, api_base=api_base)
 giskard.llm.set_embedding_model("ollama/nomic-embed-text", api_base=api_base)
 ```
 
@@ -130,7 +130,7 @@ os.environ["AWS_ACCESS_KEY_ID"] = "" # "my-aws-access-key"
 os.environ["AWS_SECRET_ACCESS_KEY"] = "" # "my-aws-secret-access-key"
 os.environ["AWS_REGION_NAME"] = "" # "us-west-2"
 
-giskard.llm.set_llm_model("bedrock/anthropic.claude-3-sonnet-20240229-v1:0")
+giskard.llm.set_llm_model("bedrock/anthropic.claude-3-sonnet-20240229-v1:0", disable_structured_output=True)
 giskard.llm.set_embedding_model("bedrock/amazon.titan-embed-image-v1")
 ```
 

diff --git a/docs/open_source/setting_up/index.md b/docs/open_source/setting_up/index.md
@@ -104,7 +104,7 @@ import giskard
 api_base = "http://localhost:11434" # default api_base for local Ollama
 
 # See supported models here: https://docs.litellm.ai/docs/providers/ollama#ollama-models
-giskard.llm.set_llm_model("ollama/llama3", disable_structured_output=True, api_base=api_base)
+giskard.llm.set_llm_model("ollama/llama3.1", disable_structured_output=True, api_base=api_base)
 giskard.llm.set_embedding_model("ollama/nomic-embed-text", api_base=api_base)
 ```
 
@@ -129,7 +129,7 @@ os.environ["AWS_ACCESS_KEY_ID"] = "" # "my-aws-access-key"
 os.environ["AWS_SECRET_ACCESS_KEY"] = "" # "my-aws-secret-access-key"
 os.environ["AWS_REGION_NAME"] = "" # "us-west-2"
 
-giskard.llm.set_llm_model("bedrock/anthropic.claude-3-sonnet-20240229-v1:0")
+giskard.llm.set_llm_model("bedrock/anthropic.claude-3-sonnet-20240229-v1:0", disable_structured_output=True)
 giskard.llm.set_embedding_model("bedrock/amazon.titan-embed-image-v1")
 ```
 

diff --git a/docs/open_source/testset_generation/testset_generation/index.md b/docs/open_source/testset_generation/testset_generation/index.md
@@ -149,7 +149,7 @@ import giskard
 api_base = "http://localhost:11434" # default api_base for local Ollama
 
 # See supported models here: https://docs.litellm.ai/docs/providers/ollama#ollama-models
-giskard.llm.set_llm_model("ollama/llama3", disable_structured_output=True, api_base=api_base)
+giskard.llm.set_llm_model("ollama/llama3.1", disable_structured_output=True, api_base=api_base)
 giskard.llm.set_embedding_model("ollama/nomic-embed-text", api_base=api_base)
 ```
 
@@ -168,7 +168,7 @@ os.environ["AWS_ACCESS_KEY_ID"] = "" # "my-aws-access-key"
 os.environ["AWS_SECRET_ACCESS_KEY"] = "" # "my-aws-secret-access-key"
 os.environ["AWS_REGION_NAME"] = "" # "us-west-2"
 
-giskard.llm.set_llm_model("bedrock/anthropic.claude-3-sonnet-20240229-v1:0")
+giskard.llm.set_llm_model("bedrock/anthropic.claude-3-sonnet-20240229-v1:0", disable_structured_output=True)
 giskard.llm.set_embedding_model("bedrock/amazon.titan-embed-image-v1")
 ```