apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-03 06:51:00 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

feat: add GLM (Zhipu AI) variant for OpenAI processor (#1009)

Browse source
This commit is contained in:
cybermaggedon 2026-07-01 16:20:43 +01:00 • committed by GitHub
parent 656ca430b9
commit 11ca7c89c4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 23 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

23
trustgraph-flow/trustgraph/model/text_completion/openai/variants.py
View file

@ -129,6 +129,28 @@ class MistralVariant(Variant):
return {"reasoning_effort": effort}
class GlmVariant(Variant):
"""GLM / Zhipu AI API (GLM-4, GLM-4.7, etc.)."""
name = "glm"
token_param = "max_tokens"
temperature_with_thinking = True
def completion_kwargs(self, max_output, temperature, thinking):
enabled = "enabled" if thinking != "off" else "disabled"
kwargs = {
self.token_param: max_output,
"temperature": temperature,
"extra_body": {
"thinking": {"type": enabled},
},
}
return kwargs
def thinking_kwargs(self, effort):
return {}
class LlamaVariant(Variant):
"""Llama models via OpenAI-compatible servers (vLLM, Ollama, etc.).
@ -159,6 +181,7 @@ VARIANTS = {
"deepseek": DeepSeekVariant,
"qwen": QwenVariant,
"mistral": MistralVariant,
"glm": GlmVariant,
"llama": LlamaVariant,
}