Merge pull request #108 from VinciGit00/93-groq-model-implementation

feat: base groq + requirements + toml update with groq

Author: VinciGit00

pyproject.toml

@@ -38,6 +38,7 @@ graphviz = "0.20.1"
 google = "3.0.0"
 minify-html = "0.15.0"
 free-proxy = "1.1.1"
+langchain-groq = "0.1.3"
 
 [tool.poetry.dev-dependencies]
 pytest = "8.0.0"

requirements.txt

@@ -11,4 +11,5 @@ python-dotenv==1.0.1
 tiktoken>=0.5.2,<0.6.0
 tqdm==4.66.1
 minify-html==0.15.0
-free-proxy==1.1.1
+langchain-groq==0.1.3
+free-proxy==1.1.1

scrapegraphai/models/groq.py

@@ -0,0 +1,21 @@
+"""
+Groq module configuration
+"""
+
+from langchain_groq import ChatGroq
+
+
+class Groq(ChatGroq):
+    """Class for wrapping Groq module"""
+
+    def __init__(self, llm_config: dict):
+        """
+        A wrapper for the Groq class that provides default configuration
+        and could be extended with additional methods if needed.
+
+        Args:
+            llm_config (dict): Configuration parameters for the language model.
+            such as model="llama3-70b-8192" and api_key
+        """
+        # Initialize the superclass (ChatOpenAI) with provided config parameters
+        super().__init__(**llm_config)