quick fix to get model given from UI

openhands/core/config2.py

@@ -0,0 +1 @@
+model = 'groq/llama3-8b-8192'

openhands/core/message.py

@@ -65,8 +65,11 @@ def contains_image(self) -> bool:
     @model_serializer
     def serialize_model(self) -> dict:
         content: list[dict[str, str | dict[str, str]]] = []
-        model = config.get_llm_config().model
+        import openhands.core.config2 as config2
+
+        model = config2.model
         supports_vision = litellm.supports_vision(model)
+
         if not supports_vision:
             text_contents = '\n'.join([item.text for item in self.content])
             return {'role': self.role, 'content': text_contents}

openhands/server/session/session.py

@@ -94,6 +94,9 @@ async def _initialize_agent(self, data: dict):
             default_llm_config.model = args.get(
                 ConfigType.LLM_MODEL, default_llm_config.model
             )
+            import openhands.core.config2 as config2
+
+            config2.model = default_llm_config.model
             default_llm_config.api_key = args.get(
                 ConfigType.LLM_API_KEY, default_llm_config.api_key
             )