Skip to content

Commit

Permalink
fixing docs for multiple ai extraction providers
Browse files Browse the repository at this point in the history
  • Loading branch information
@himynamesdave
himynamesdave committed Nov 11, 2024
1 parent 6307662 commit 2d0db8d
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 17 deletions.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ How the extractions are performed
If any AI extractions, or AI relationship mode is set, you must set the following accordingly

* `--ai_settings_extractions`:
* defines the `provider:model` to be used for extractions. You can supply more than one provider. If more than one provider passed, txt2stix will take extractions from all models, de-dupelicate them, and them package them in the output. Currently supports:
* defines the `provider:model` to be used for extractions. You can supply more than one provider. Seperate with a space (e.g. `gpt-4o claude-3-opus-latest`) If more than one provider passed, txt2stix will take extractions from all models, de-dupelicate them, and them package them in the output. Currently supports:
* Provider: `openai:`, models e.g.: `gpt-4o`, `gpt-4o-mini`, `gpt-4-turbo`, `gpt-4` ([More here](https://platform.openai.com/docs/models))
* Provider: `anthropic:`, models e.g.: `claude-3-5-sonnet-latest`, `claude-3-5-haiku-latest`, `claude-3-opus-latest` ([More here](https://docs.anthropic.com/en/docs/about-claude/models))
* Provider: `gemini:models/`, models: `gemini-1.5-pro-latest`, `gemini-1.5-flash-latest` ([More here](https://ai.google.dev/gemini-api/docs/models/gemini))
Expand Down
32 changes: 16 additions & 16 deletions tests/manual-tests/cases-ai-extraction-type.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_btc_wallet \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.2 ai_cryptocurrency_btc_transaction
Expand All @@ -21,7 +21,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_btc_transaction \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.3 ai_cryptocurrency_eth_wallet
Expand All @@ -34,7 +34,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_eth_wallet \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.4 ai_cryptocurrency_eth_transaction
Expand All @@ -47,7 +47,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_eth_transaction \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.5 ai_cryptocurrency_xmr_wallet
Expand All @@ -60,7 +60,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_xmr_wallet \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.6 ai_cryptocurrency_xmr_transaction
Expand All @@ -73,7 +73,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_xmr_transaction \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.7 ai_phone_number
Expand All @@ -86,7 +86,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_phone_number \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.8 ai_country_alpha2
Expand All @@ -99,7 +99,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_country_alpha2 \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.9 ai_mitre_attack_enterprise
Expand All @@ -112,7 +112,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_mitre_attack_enterprise \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.10 ai_mitre_attack_mobile
Expand All @@ -125,7 +125,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_mitre_attack_mobile \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.11 ai_mitre_attack_ics
Expand All @@ -138,7 +138,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_mitre_attack_ics \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.12 ai_mitre_capec
Expand All @@ -151,7 +151,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_mitre_capec \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

#### 1.1.13 ai_mitre_cwe
Expand All @@ -164,7 +164,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_mitre_cwe \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```


Expand All @@ -179,7 +179,7 @@ python3 txt2stix.py \
--confidence 100 \
--use_whitelist examples_whitelist \
--use_extractions ai_mitre_attack_enterprise \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

Expect 0 extractions.
Expand All @@ -192,7 +192,7 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_mitre_attack_enterprise \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

* course-of-action (2)
Expand All @@ -212,5 +212,5 @@ python3 txt2stix.py \
--tlp_level clear \
--confidence 100 \
--use_extractions ai_cryptocurrency_btc_wallet,ai_cryptocurrency_btc_transaction,ai_cryptocurrency_eth_wallet,ai_cryptocurrency_eth_transaction,ai_cryptocurrency_xmr_wallet,ai_cryptocurrency_xmr_transaction,ai_phone_number,ai_country_alpha2,ai_mitre_attack_enterprise,ai_mitre_attack_mobile,ai_mitre_attack_ics,ai_mitre_capec,ai_mitre_cwe \
--ai_settings_extractions openai:gpt-4o,anthropic:claude-3-5-sonnet-latest,gemini:models/gemini-1.5-pro-latest
--ai_settings_extractions openai:gpt-4o anthropic:claude-3-5-sonnet-latest gemini:models/gemini-1.5-pro-latest
```

0 comments on commit 2d0db8d

Please sign in to comment.