Skip to content

Commit

Permalink
fix: python script
Browse files Browse the repository at this point in the history
  • Loading branch information
sangjanai committed Jun 5, 2024
1 parent f7b46e3 commit e59313e
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions .github/scripts/e2e-test-server.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,16 +9,14 @@

n = len(sys.argv)
print("Total arguments passed:", n)
if n < 4:
print("The number of arguments should >= 4")
if n < 3:
print("The number of arguments should >= 3")
exit(1)

BINARY_PATH = sys.argv[1]
if platform.system == 'Windows':
BINARY_PATH += '.exe'
MODEL_PATH = sys.argv[2]
EMBED_MODEL = 'embedding_model'
EMBED_FILE_TO_SAVE = './' + EMBED_MODEL + '.gguf'

CONST_CTX_SIZE = 1024
CONST_USER_ROLE = "user"
Expand Down Expand Up @@ -97,8 +95,6 @@ def StopServer():
def CleanUp():
StopServer()
p.communicate()
os.remove(LLM_FILE_TO_SAVE)
os.remove(EMBED_FILE_TO_SAVE)
with open('./test.log', 'r') as f:
print(f.read())

Expand Down

0 comments on commit e59313e

Please sign in to comment.