Remove old api methods for managing clusters.

Author: inf3rnus

langchain_bytez/BytezModelMixin.py

@@ -553,20 +553,6 @@ async def _acall(
 
         return text
 
-    def start_cluster(self):
-        payload = {"capacity": self.capacity, "timeout": self.timeout}
-        return self._make_request(method="put", payload=payload)
-
-    def get_cluster(self):
-        return self._make_request(method="get")
-
-    def update_cluster(self):
-        payload = {"capacity": self.capacity, "timeout": self.timeout}
-        return self._make_request(method="patch", payload=payload)
-
-    def shutdown_cluster(self):
-        return self._make_request(method="delete")
-
 
 def _convert_message_to_dict(message: BaseMessage) -> dict:
     """Convert a LangChain message to a dictionary.

langchain_bytez/test.py

@@ -78,42 +78,6 @@ async def run_async_test():
         bytez_chat_model_llama,
     ]
 
-    def shutdown_models():
-        for model in models:
-            try:
-                model.shutdown_cluster()
-            except Exception as exception:
-                print(f"{model.model_id}", exception)
-
-    def start_models():
-        for model in models:
-            try:
-                model.start_cluster()
-            except Exception as exception:
-                print(f"{model.model_id}", exception)
-
-            result = model.get_cluster()
-            status = result["status"]
-
-            while result["status"] == "loading":
-                result = model.get_cluster()
-                status = result["status"]
-
-                print(f"{model.model_id} status: {status}")
-
-                time.sleep(5)
-
-            if status == "on":
-                continue
-
-            raise Exception(
-                f"{model.model_id} - Cluster failed to create with status: {status}"
-            )
-
-    start_models()
-
-    # shutdown_models()
-
     tests = [
         #
         # interface tests
@@ -160,7 +124,6 @@ def start_models():
 
     print_text("\n\n>>>>>>>>>> Tests pass <<<<<<<<<<<", color="blue")
 
-    # shutdown_models()
     pass