diff options
Diffstat (limited to 'continuedev')
| -rw-r--r-- | continuedev/main.py | 5 | ||||
| -rw-r--r-- | continuedev/src/continuedev/libs/util/count_tokens.py | 1 | ||||
| -rw-r--r-- | continuedev/src/run.py | 4 |
3 files changed, 6 insertions, 4 deletions
diff --git a/continuedev/main.py b/continuedev/main.py new file mode 100644 index 00000000..3cf4e817 --- /dev/null +++ b/continuedev/main.py @@ -0,0 +1,5 @@ +from .src.continuedev.server.main import run_server + + +def main(): + run_server() diff --git a/continuedev/src/continuedev/libs/util/count_tokens.py b/continuedev/src/continuedev/libs/util/count_tokens.py index 6add7b1a..3b594036 100644 --- a/continuedev/src/continuedev/libs/util/count_tokens.py +++ b/continuedev/src/continuedev/libs/util/count_tokens.py @@ -3,6 +3,7 @@ from typing import Dict, List, Union from ...core.main import ChatMessage from .templating import render_templated_string from ...libs.llm import LLM +from tiktoken_ext import openai_public import tiktoken # TODO move many of these into specific LLM.properties() function that diff --git a/continuedev/src/run.py b/continuedev/src/run.py deleted file mode 100644 index 089cc54d..00000000 --- a/continuedev/src/run.py +++ /dev/null @@ -1,4 +0,0 @@ -from continuedev.server.main import run_server - -if __name__ == "__main__": - run_server() |