Merge branch 'main' of https://github.com/mtaruno/SWE-Agent-codegeex

mtaruno · Sep 16, 2024 · 814810d · 814810d
2 parents 7caea70 + 16a48b0
commit 814810d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/sweagent/agent/models.py b/sweagent/agent/models.py
@@ -1210,6 +1210,11 @@ def get_model(args: ModelArguments, commands: list[Command] | None = None):
         return GroqModel(args, commands)
     elif args.model_name == "instant_empty_submit":
         return InstantEmptySubmitTestModel(args, commands)
+    elif args.model_name.startswith("glm"):
+        return GLMModel(args, commands)
+    elif args.model_name.startswith("codegeex"):
+        logger.info("Entering CodeGeeX Model") 
+        return codegeexModel(args, commands)
     else:
         msg = f"Invalid model name: {args.model_name}"
         raise ValueError(msg)
diff --git a/sweagent/environment/swe_env.py b/sweagent/environment/swe_env.py
@@ -15,6 +15,7 @@
 from pathlib import Path
 from typing import Any
 import tempfile
+import tempfile
 
 import gymnasium as gym
 import yaml