Auto-sync from demo at Thu Nov 20 11:09:35 UTC 2025

app.py

@@ -42,7 +42,7 @@ def init_graph_gen(config: dict, env: dict) -> GraphGen:
 
     tokenizer_instance = Tokenizer(config.get("tokenizer", "cl100k_base"))
     synthesizer_llm_client = OpenAIClient(
-        model_name=env.get("SYNTHESIZER_MODEL", ""),
+        model=env.get("SYNTHESIZER_MODEL", ""),
         base_url=env.get("SYNTHESIZER_BASE_URL", ""),
         api_key=env.get("SYNTHESIZER_API_KEY", ""),
         request_limit=True,
@@ -51,7 +51,7 @@ def init_graph_gen(config: dict, env: dict) -> GraphGen:
         tokenizer=tokenizer_instance,
     )
     trainee_llm_client = OpenAIClient(
-        model_name=env.get("TRAINEE_MODEL", ""),
+        model=env.get("TRAINEE_MODEL", ""),
         base_url=env.get("TRAINEE_BASE_URL", ""),
         api_key=env.get("TRAINEE_API_KEY", ""),
         request_limit=True,

graphgen/models/llm/api/openai_client.py

@@ -32,7 +32,7 @@ class OpenAIClient(BaseLLMWrapper):
     def __init__(
         self,
         *,
-        model_name: str = "gpt-4o-mini",
+        model: str = "gpt-4o-mini",
         api_key: Optional[str] = None,
         base_url: Optional[str] = None,
         json_mode: bool = False,
@@ -44,7 +44,7 @@ class OpenAIClient(BaseLLMWrapper):
         **kwargs: Any,
     ):
         super().__init__(**kwargs)
-        self.model_name = model_name
+        self.model = model
         self.api_key = api_key
         self.base_url = base_url
         self.json_mode = json_mode
@@ -109,7 +109,7 @@ class OpenAIClient(BaseLLMWrapper):
         kwargs["max_tokens"] = 1
 
         completion = await self.client.chat.completions.create(  # pylint: disable=E1125
-            model=self.model_name, **kwargs
+            model=self.model, **kwargs
         )
 
         tokens = get_top_response_tokens(completion)
@@ -141,7 +141,7 @@ class OpenAIClient(BaseLLMWrapper):
             await self.tpm.wait(estimated_tokens, silent=True)
 
         completion = await self.client.chat.completions.create(  # pylint: disable=E1125
-            model=self.model_name, **kwargs
+            model=self.model, **kwargs
         )
         if hasattr(completion, "usage"):
             self.token_usage.append(

webui/app.py

@@ -42,7 +42,7 @@ def init_graph_gen(config: dict, env: dict) -> GraphGen:
 
     tokenizer_instance = Tokenizer(config.get("tokenizer", "cl100k_base"))
     synthesizer_llm_client = OpenAIClient(
-        model_name=env.get("SYNTHESIZER_MODEL", ""),
+        model=env.get("SYNTHESIZER_MODEL", ""),
         base_url=env.get("SYNTHESIZER_BASE_URL", ""),
         api_key=env.get("SYNTHESIZER_API_KEY", ""),
         request_limit=True,
@@ -51,7 +51,7 @@ def init_graph_gen(config: dict, env: dict) -> GraphGen:
         tokenizer=tokenizer_instance,
     )
     trainee_llm_client = OpenAIClient(
-        model_name=env.get("TRAINEE_MODEL", ""),
+        model=env.get("TRAINEE_MODEL", ""),
         base_url=env.get("TRAINEE_BASE_URL", ""),
         api_key=env.get("TRAINEE_API_KEY", ""),
         request_limit=True,