dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-27 15:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

wip

Browse source
This commit is contained in:
Paul Gauthier 2024-02-08 07:18:39 -08:00
parent 61531b9430
commit 1758937042
3 changed files with 157 additions and 91 deletions
Download patch file Download diff file Expand all files Collapse all files

30
aider/commands.py
View file

@ -1,9 +1,3 @@
from pathlib import Path
from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager
import re
import subprocess
import sys
@ -13,7 +7,7 @@ import git
from prompt_toolkit.completion import Completion
from aider import prompts, voice
from aider.utils import is_gpt4_with_openai_base_url, is_image_file
from aider.utils import is_gpt4_with_openai_base_url, is_image_file, scrape
from .dump import dump # noqa: F401
@ -31,18 +25,6 @@ class Commands:
self.voice_language = voice_language
self.tokenizer = coder.main_model.tokenizer
self.initialize_web_driver()
def initialize_web_driver(self):
chrome_options = Options()
chrome_options.add_argument("--headless")
chrome_options.add_argument("--disable-gpu")
chrome_options.add_argument("--no-sandbox")
chrome_options.add_argument("--disable-dev-shm-usage")
self.web_driver = webdriver.Chrome(
service=Service(ChromeDriverManager().install()),
options=chrome_options
)
def cmd_web(self, args):
"Use headless selenium to scrape a webpage and add the content to the chat"
@ -51,13 +33,9 @@ class Commands:
self.io.tool_error("Please provide a URL to scrape.")
return
try:
self.web_driver.get(url)
page_content = self.web_driver.find_element(By.TAG_NAME, "body").text
self.io.tool_output(f"Content from {url}:\n{page_content}")
return page_content
except Exception as e:
self.io.tool_error(f"Error scraping {url}: {e}")
content = scrape(url)
print(content)
return content
def is_command(self, inp):
return inp[0] in "/!"