redcodegen 0.0.2__tar.gz → 0.0.3__tar.gz

{redcodegen-0.0.2 → redcodegen-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: redcodegen
-Version: 0.0.2
+Version: 0.0.3
 Summary: Add your description here
 Requires-Dist: click>=8.0.0
 Requires-Dist: cwe2>=3.0.0

{redcodegen-0.0.2 → redcodegen-0.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "redcodegen"
-version = "0.0.2"
+version = "0.0.3"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.11"

{redcodegen-0.0.2 → redcodegen-0.0.3}/redcodegen/constants.py

@@ -11,7 +11,7 @@ CWE_TOP_25 = [
     798, 190, 400, 306
 ]
 
-def create_lm(model_name="openai/gpt-4o-mini", api_key=None):
+def create_lm(model_name="openai/gpt-4o-mini", temperature=0.8, api_key=None):
     """Create a DSPy language model instance.
 
     Args:
@@ -23,7 +23,12 @@ def create_lm(model_name="openai/gpt-4o-mini", api_key=None):
     """
     if api_key is None:
         api_key = os.environ.get("OPENAI_API_KEY")
-    return dspy.LM(model_name, api_key=api_key)
+    return dspy.LM(
+        model_name,
+        api_key=api_key,
+        temperature=temperature,
+        max_tokens=16000
+    )
 
 SCENARIO_EXAMPLES = [
     dspy.Example(

redcodegen-0.0.3/redcodegen/data/__init__.py

@@ -0,0 +1,2 @@
+
+m 

{redcodegen-0.0.2 → redcodegen-0.0.3}/redcodegen/main.py

@@ -155,7 +155,13 @@ def append_to_jsonl(record: Dict[str, Any], output_path: Path):
     default=None,
     help='API key (defaults to OPENAI_API_KEY env var)'
 )
-def main(cwes, use_top_25, min_samples, output, model, api_key):
+@click.option(
+    '--temperature',
+    default=0.8,
+    type=float,
+    help='Temperature for code generation (default: 0.8)'
+)
+def main(cwes, use_top_25, min_samples, output, model, api_key, temperature):
     """Generate and evaluate vulnerable code samples for specified CWEs.
 
     Examples:
@@ -166,7 +172,7 @@ def main(cwes, use_top_25, min_samples, output, model, api_key):
         python -m redcodegen --use-top-25 --model openai/gpt-4o # switch model
     """
     # Configure DSPy with specified model
-    lm = create_lm(model_name=model, api_key=api_key)
+    lm = create_lm(model_name=model, temperature=temperature, api_key=api_key)
     dspy.configure(lm=lm)
     logger.info(f"Configured model: {model}")
 

redcodegen-0.0.2/redcodegen/#seeds.py#

@@ -1,17 +0,0 @@
-import dspy
-import jsonlines
-from cwe2.database import Database
-
-from redcodegen.constants import LM
-
-dspy.configure(lm=LM)
-
-
-db = Database()
-entry = db.get(502)
-
-print(entry.name)
-print(entry.extended_description)
-
-
-