fix model_choice param

translator/agent.py

@@ -1,19 +1,31 @@
 from .parse import parse_csharp
 from .mapper import load_mapping
 from .synth import synthesize_cpp, synthesize_blueprint_python
+from .llm_client import generate_with_hf, generate_with_openai
+
 
 class TranslationAgent:
     def __init__(self, mapping_path="mapping_table.json"):
         self.mapping = load_mapping(mapping_path)
 
-    def translate(self, csharp_code: str, target_format="both", backend="hf"):
+    def translate(self, csharp_code: str, target_format="both", model_choice="hf"):
+        """
+        Convert Unity C# code into Unreal C++ and/or Blueprint Python.
+
+        Args:
+            csharp_code: Unity source
+            target_format: "cpp", "blueprint", or "both"
+            model_choice: "hf" or "openai"
+        """
         parsed = parse_csharp(csharp_code)
         results = {"cpp": "", "blueprint_py": ""}
 
-        if target_format in ("cpp", "both"):
-            results["cpp"] = synthesize_cpp(parsed, self.mapping, backend=backend)
+        # choose model backend
+        backend = "hf" if model_choice == "hf" else "openai"
 
+        if target_format in ("cpp", "both"):
+            results["cpp"] = synthesize_cpp(parsed, self.mapping, model_backend=backend)
         if target_format in ("blueprint", "both"):
-            results["blueprint_py"] = synthesize_blueprint_python(parsed, self.mapping, backend=backend)
+            results["blueprint_py"] = synthesize_blueprint_python(parsed, self.mapping, model_backend=backend)
 
         return results["cpp"], results["blueprint_py"]

translator/synth.py

@@ -1,6 +1,6 @@
 from .llm_client import generate_with_hf, generate_with_openai
 
-def synthesize_cpp(parsed: dict, mapping: dict, backend: str = "hf", prompt_context: str = "") -> str:
+def synthesize_cpp(parsed: dict, mapping: dict, prompt_context: str = "", model_backend="hf") -> str:
     """
     Converts Unity C# parsed structure into Unreal Engine C++ (header + source).
     """
@@ -19,12 +19,15 @@ def synthesize_cpp(parsed: dict, mapping: dict, backend: str = "hf", prompt_cont
     {prompt_context}
     """
 
-    if backend == "hf":
-        return generate_with_hf(prompt)
+    if model_backend == "openai":
+        from .llm_client import generate_with_openai
+        return generate_with_openai(prompt, model="gpt-4o-mini")
     else:
-        return generate_with_openai(prompt)
+        from .llm_client import generate_with_hf
+        return generate_with_hf(prompt, model="mistralai/Mixtral-8x7B-Instruct")
 
-def synthesize_blueprint_python(parsed: dict, mapping: dict, backend: str = "hf") -> str:
+
+def synthesize_blueprint_python(parsed: dict, mapping: dict, model_backend="hf") -> str:
     prompt = f"""
     Given this parsed Unity class and mapping rules, produce a Python Editor script using Unreal's Python API
     that will create a Blueprint Actor in /Game/Blueprints named BP_<ClassName> with:
@@ -38,7 +41,9 @@ def synthesize_blueprint_python(parsed: dict, mapping: dict, backend: str = "hf"
     Mapping: {mapping}
     """
 
-    if backend == "hf":
-        return generate_with_hf(prompt)
+    if model_backend == "openai":
+        from .llm_client import generate_with_openai
+        return generate_with_openai(prompt, model="gpt-4o-mini")
     else:
-        return generate_with_openai(prompt)
+        from .llm_client import generate_with_hf
+        return generate_with_hf(prompt, model="mistralai/Mixtral-8x7B-Instruct")