summaryrefslogtreecommitdiff
path: root/continuedev
diff options
context:
space:
mode:
authorNate Sesti <sestinj@gmail.com>2023-07-31 22:27:08 -0700
committerNate Sesti <sestinj@gmail.com>2023-07-31 22:27:08 -0700
commite18df47cee14cde58584ffc1a8bbb2606b6f4102 (patch)
tree9f5967358100ed5b729afb49883bc5e7d5f63c05 /continuedev
parentfa221af037ced05018b6441575b7e22b5d6902a8 (diff)
downloadsncontinue-e18df47cee14cde58584ffc1a8bbb2606b6f4102.tar.gz
sncontinue-e18df47cee14cde58584ffc1a8bbb2606b6f4102.tar.bz2
sncontinue-e18df47cee14cde58584ffc1a8bbb2606b6f4102.zip
ci: :green_heart: fix file structure for pyinstaller, main.yaml updates
Diffstat (limited to 'continuedev')
-rw-r--r--continuedev/main.py5
-rw-r--r--continuedev/src/continuedev/libs/util/count_tokens.py1
-rw-r--r--continuedev/src/run.py4
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()