fix: :bug: number of bug fixes

4 files changed, 15 insertions, 7 deletions

diff --git a/continuedev/src/continuedev/libs/llm/hf_tgi.py b/continuedev/src/continuedev/libs/llm/hf_tgi.py
index f106f83f..5c7e0239 100644
--- a/continuedev/src/continuedev/libs/llm/hf_tgi.py
+++ b/continuedev/src/continuedev/libs/llm/hf_tgi.py
@@ -52,7 +52,7 @@ class HuggingFaceTGI(LLM):
         ) as client_session:
             async with client_session.post(
                 f"{self.server_url}",
-                json={"inputs": prompt, **self._transform_args(args)},
+                json={"inputs": prompt, **args},
             ) as resp:
                 async for line in resp.content.iter_any():
                     if line:
diff --git a/continuedev/src/continuedev/libs/llm/llamacpp.py b/continuedev/src/continuedev/libs/llm/llamacpp.py
index 7940c4c9..84eb85fd 100644
--- a/continuedev/src/continuedev/libs/llm/llamacpp.py
+++ b/continuedev/src/continuedev/libs/llm/llamacpp.py
@@ -87,7 +87,7 @@ class LlamaCpp(LLM):
                 f"{self.server_url}/completion",
                 json={
                     "prompt": prompt,
-                    **self._transform_args(args),
+                    **args,
                 },
                 headers={"Content-Type": "application/json"},
             ) as resp:
@@ -107,7 +107,7 @@ class LlamaCpp(LLM):
             self.model,
             messages,
             self.context_length,
-            args["max_tokens"],
+            args["n_predict"] if "n_predict" in args else 1024,
             None,
             functions=args.get("functions", None),
             system_message=self.system_message,
@@ -123,7 +123,7 @@ class LlamaCpp(LLM):
             ) as client_session:
                 async with client_session.post(
                     f"{self.server_url}/completion",
-                    json={"prompt": prompt, **self._transform_args(args)},
+                    json={"prompt": prompt, **args},
                     headers=headers,
                 ) as resp:
                     async for line in resp.content:
@@ -170,7 +170,7 @@ class LlamaCpp(LLM):
             ) as client_session:
                 async with client_session.post(
                     f"{self.server_url}/completion",
-                    json={"prompt": prompt, **self._transform_args(args)},
+                    json={"prompt": prompt, **args},
                     headers={"Content-Type": "application/json"},
                 ) as resp:
                     json_resp = await resp.json()
diff --git a/continuedev/src/continuedev/plugins/context_providers/file.py b/continuedev/src/continuedev/plugins/context_providers/file.py
index 859088b8..a300a4a9 100644
--- a/continuedev/src/continuedev/plugins/context_providers/file.py
+++ b/continuedev/src/continuedev/plugins/context_providers/file.py
@@ -123,7 +123,7 @@ class FileContextProvider(ContextProvider):
         )
 
     async def provide_context_items(self, workspace_dir: str) -> List[ContextItem]:
-        contents = await self.sdk.ide.listDirectoryContents(workspace_dir, True)
+        contents = await self.sdk.ide.listDirectoryContents(workspace_dir, False)
         if contents is None:
             return []
 
diff --git a/extension/src/continueIdeClient.ts b/extension/src/continueIdeClient.ts
index 539eb10d..582caa53 100644
--- a/extension/src/continueIdeClient.ts
+++ b/extension/src/continueIdeClient.ts
@@ -72,7 +72,8 @@ class IdeProtocolClient {
       this.handleMessage(messageType, data, messenger).catch((err) => {
         vscode.window
           .showErrorMessage(
-            "Error handling message from Continue server: " + err.message,
+            `Error handling message (${messageType}) from Continue server: ` +
+              err.message,
             "View Logs"
           )
           .then((selection) => {
@@ -613,6 +614,13 @@ class IdeProtocolClient {
     let contents: string | undefined;
     if (typeof contents === "undefined") {
       try {
+        const fileStats = await vscode.workspace.fs.stat(
+          uriFromFilePath(filepath)
+        );
+        if (fileStats.size > 1000000) {
+          return "";
+        }
+
         contents = await vscode.workspace.fs
           .readFile(uriFromFilePath(filepath))
           .then((bytes) => new TextDecoder().decode(bytes));