fix(ollama): add shorthand support and fix test configuration

- Add "ollama::" to SHORTHAND_MODEL_PREFIXES for shorthand model syntax
- Add ollama case in from_shorthand match for shorthand resolution
- Quote variant names with dots in TOML configs to prevent parsing issues
- Use correct model name (qwen2:0.5b) for shorthand test variant

Author: rrrodzilla

tensorzero-core/src/model.rs

@@ -2323,6 +2323,7 @@ pub const SHORTHAND_MODEL_PREFIXES: &[&str] = &[
     "hyperbolic::",
     "groq::",
     "mistral::",
+    "ollama::",
     "openai::",
     "openrouter::",
     "together::",
@@ -2440,6 +2441,13 @@ impl ShorthandModelConfig for ModelConfig {
                     .get_defaulted_credential(None, default_credentials)
                     .await?,
             )),
+            "ollama" => ProviderConfig::Ollama(OllamaProvider::new(
+                model_name,
+                None,
+                OllamaKind
+                    .get_defaulted_credential(None, default_credentials)
+                    .await?,
+            )),
             #[cfg(any(test, feature = "e2e_tests"))]
             "dummy" => ProviderConfig::Dummy(DummyProvider::new(model_name, None)?),
             _ => {

tensorzero-core/tests/e2e/config/tensorzero.functions.basic_test.toml

@@ -792,14 +792,14 @@ max_tokens = 100
 
 ## Ollama
 
-[functions.basic_test.variants.dolphin-llama3-8b-ollama]
+[functions.basic_test.variants."qwen2-0.5b-ollama"]
 type = "chat_completion"
-model = "dolphin-llama3-8b-ollama"
+model = "qwen2-0.5b-ollama"
 system_template = "../../../fixtures/config/functions/basic_test/prompt/system_template.minijinja"
 max_tokens = 100
 
 [functions.basic_test.variants.ollama-shorthand]
 type = "chat_completion"
-model = "ollama::dolphin-llama3:8b"
+model = "ollama::qwen2:0.5b"
 system_template = "../../../fixtures/config/functions/basic_test/prompt/system_template.minijinja"
 max_tokens = 100

tensorzero-core/tests/e2e/config/tensorzero.functions.json_success.toml

@@ -734,9 +734,9 @@ max_tokens = 100
 
 ## Ollama
 
-[functions.json_success.variants.dolphin-llama3-8b-ollama]
+[functions.json_success.variants."qwen2-0.5b-ollama"]
 type = "chat_completion"
-model = "dolphin-llama3-8b-ollama"
+model = "qwen2-0.5b-ollama"
 system_template = "../../../fixtures/config/functions/json_success/prompt/system_template.minijinja"
 user_template = "../../../fixtures/config/functions/json_success/prompt/user_template.minijinja"
 json_mode = "on"