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__":