feat: :sparkles: testing in ci, final test of

2 files changed, 25 insertions, 30 deletions

diff --git a/.github/workflows/main.yaml b/.github/workflows/main.yaml
index 6111856d..9bc05281 100644
--- a/.github/workflows/main.yaml
+++ b/.github/workflows/main.yaml
@@ -45,7 +45,7 @@ jobs:
           name: ${{ runner.os }} Build
           path: dist/*
 
-  test-and-package:
+  test-and-publish-extension:
     needs: pyinstaller
     strategy:
       matrix:
@@ -135,51 +135,42 @@ jobs:
           npm run test
         if: matrix.os != 'ubuntu-20.04'
 
-      # Upload .vsix artifact
-
-      - name: Upload .vsix as an artifact
-        uses: actions/upload-artifact@v2
-        with:
-          name: vsix-artifact
-          path: extension/build/*
-        if: matrix.os == 'ubuntu-20.04'
-
-  publish:
-    needs: test-and-package
-    runs-on: ubuntu-20.04
-    permissions:
-      contents: write
-
-    steps:
-      # Checkout and download .vsix artifact
-
-      - name: Checkout
-        uses: actions/checkout@v2
-
-      - name: Download .vsix artifact
-        uses: actions/download-artifact@v2
-        with:
-          name: vsix-artifact
-          path: extension/build
-
-      # Publish the extension and commit/push the version change
+      # Publish the extension and commit/push the version change (ONLY on ubuntu-20.04)
 
       - name: Publish
         run: |
           cd extension
           npx vsce publish patch -p ${{ secrets.VSCE_TOKEN }}
+        if: matrix.os == 'ubuntu-20.04'
+
       - name: Commit changes
         run: |
           git config --local user.email "action@github.com"
           git config --local user.name "GitHub Action"
           git commit -am "ci: 💚 Update package.json version [skip ci]"
+        if: matrix.os == 'ubuntu-20.04'
 
       - name: Push changes
         uses: ad-m/github-push-action@master
         with:
           github_token: ${{ secrets.GITHUB_TOKEN }}
           branch: ${{ github.ref }}
+        if: matrix.os == 'ubuntu-20.04'
+
+      - name: Upload .vsix as an artifact
+        uses: actions/upload-artifact@v2
+        with:
+          name: vsix-artifact
+          path: extension/build/*
+        if: matrix.os == 'ubuntu-20.04'
 
+  publish-binaries:
+    needs: test-and-publish-extension
+    runs-on: ubuntu-20.04
+    permissions:
+      contents: write
+
+    steps:
       # Download binaries and upload to S3
 
       - name: Download Linux build
diff --git a/continuedev/src/continuedev/libs/llm/together.py b/continuedev/src/continuedev/libs/llm/together.py
index 1cc0a711..c3f171c9 100644
--- a/continuedev/src/continuedev/libs/llm/together.py
+++ b/continuedev/src/continuedev/libs/llm/together.py
@@ -49,6 +49,8 @@ class TogetherLLM(LLM):
             prompt += f"<human>: Hi!\n<bot>: {system_message}\n"
         for message in chat_messages:
             prompt += f'<{"human" if message["role"] == "user" else "bot"}>: {message["content"]}\n'
+
+        prompt += "<bot>:"
         return prompt
 
     async def stream_complete(self, prompt, with_history: List[ChatMessage] = None, **kwargs) -> Generator[Union[Any, List, Dict], None, None]:
@@ -113,6 +115,8 @@ class TogetherLLM(LLM):
             "Authorization": f"Bearer {self.api_key}"
         }) as resp:
             try:
-                return await resp.text()
+                text = await resp.text()
+                j = json.loads(text)
+                return j["output"]["choices"][0]["text"]
             except:
                 raise Exception(await resp.text())