Skip to content

Commit ce8cffe

Browse files
committed
fix: update main function for scripts
1 parent 5340150 commit ce8cffe

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

codegen-examples/examples/codegen-mcp-server/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,6 @@ dependencies = [
1515
]
1616

1717
[tool.poetry.scripts]
18-
codegen-mcp-server = "server"
18+
codegen-mcp-server = "server:main"
1919
[project.scripts]
20-
codegen-mcp-server = "server"
20+
codegen-mcp-server = "server:main"

codegen-examples/examples/codegen-mcp-server/server.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,8 +82,12 @@ async def execute_codemod(codemod: Annotated[str, "The python codemod code to ex
8282
return {"error": f"Error executing codemod: {str(e)}", "details": {"type": type(e).__name__, "message": str(e)}}
8383

8484

85-
if __name__ == "__main__":
85+
def main():
8686
print("starting codegen-mcp-server")
8787
run = mcp.run_stdio_async()
8888
print("codegen-mcp-server started")
8989
asyncio.run(run)
90+
91+
92+
if __name__ == "__main__":
93+
main()

0 commit comments

Comments
 (0)