From f7bf917baf17cd4d132c220966a47864f9fabcbc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciej=20Pi=C3=B3ro?= <70006947+maciek-pioro@users.noreply.github.com> Date: Sun, 16 Apr 2023 01:05:11 +0200 Subject: [PATCH] Fix missing docker-compose (#2899) Fix missing `docker-compose` command if only `docker compose` (note space) is available. --- langchain/server.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/langchain/server.py b/langchain/server.py index 4b00a478..16eb3f31 100644 --- a/langchain/server.py +++ b/langchain/server.py @@ -1,4 +1,5 @@ """Script to run langchain-server locally using docker-compose.""" +import shutil import subprocess from pathlib import Path @@ -6,8 +7,14 @@ from pathlib import Path def main() -> None: """Run the langchain server locally.""" p = Path(__file__).absolute().parent / "docker-compose.yaml" - subprocess.run(["docker-compose", "-f", str(p), "pull"]) - subprocess.run(["docker-compose", "-f", str(p), "up"]) + + if shutil.which("docker-compose") is None: + docker_compose_command = ["docker", "compose"] + else: + docker_compose_command = ["docker-compose"] + + subprocess.run([*docker_compose_command, "-f", str(p), "pull"]) + subprocess.run([*docker_compose_command, "-f", str(p), "up"]) if __name__ == "__main__":