diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/src/continuedev/__main__.py | 16 | ||||
| -rw-r--r-- | continuedev/src/continuedev/core/autopilot.py | 6 | ||||
| -rw-r--r-- | continuedev/src/continuedev/server/main.py | 30 |
3 files changed, 39 insertions, 13 deletions
diff --git a/continuedev/src/continuedev/__main__.py b/continuedev/src/continuedev/__main__.py new file mode 100644 index 00000000..6dd03046 --- /dev/null +++ b/continuedev/src/continuedev/__main__.py @@ -0,0 +1,16 @@ +import argparse + +from .server.main import run_server + + +def main(): + parser = argparse.ArgumentParser() + parser.add_argument("-p", "--port", help="server port", type=int, default=65432) + parser.add_argument("--host", help="server host", type=str, default="127.0.0.1") + args = parser.parse_args() + + run_server(port=args.port, host=args.host) + + +if __name__ == "__main__": + main() diff --git a/continuedev/src/continuedev/core/autopilot.py b/continuedev/src/continuedev/core/autopilot.py index cbf9ad59..5f17b4d2 100644 --- a/continuedev/src/continuedev/core/autopilot.py +++ b/continuedev/src/continuedev/core/autopilot.py @@ -525,6 +525,12 @@ class Autopilot(ContinueBaseModel): if self.session_info is None: async def create_title(): + if ( + self.session_info is not None + and self.session_info.title is not None + ): + return + title = await self.continue_sdk.models.medium.complete( f'Give a short title to describe the current chat session. Do not put quotes around the title. The first message was: "{user_input}". Do not use more than 10 words. The title is: ', max_tokens=20, diff --git a/continuedev/src/continuedev/server/main.py b/continuedev/src/continuedev/server/main.py index 00ded6f1..09adfa51 100644 --- a/continuedev/src/continuedev/server/main.py +++ b/continuedev/src/continuedev/server/main.py @@ -34,18 +34,8 @@ def health(): return {"status": "ok"} -try: - # add cli arg for server port - parser = argparse.ArgumentParser() - parser.add_argument("-p", "--port", help="server port", type=int, default=65432) - args = parser.parse_args() -except Exception as e: - logger.debug(f"Error parsing command line arguments: {e}") - raise e - - -def run_server(): - config = uvicorn.Config(app, host="127.0.0.1", port=args.port) +def run_server(port: int = 65432, host: str = "127.0.0.1"): + config = uvicorn.Config(app, host=host, port=port) server = uvicorn.Server(config) server.run() @@ -66,7 +56,21 @@ atexit.register(cleanup) if __name__ == "__main__": try: - run_server() + try: + # add cli arg for server port + parser = argparse.ArgumentParser() + parser.add_argument( + "-p", "--port", help="server port", type=int, default=65432 + ) + parser.add_argument( + "-h", "--host", help="server host", type=str, default="127.0.0.1" + ) + args = parser.parse_args() + except Exception as e: + logger.debug(f"Error parsing command line arguments: {e}") + raise e + + run_server(args.port, args.host) except Exception as e: logger.debug(f"Error starting Continue server: {e}") cleanup() |