AtHeartEngineer/LucidLoanMachine
1
0
Fork 0
mirror of https://github.com/hackertron/LucidLoanMachine.git synced 2026-01-09 04:28:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

updated the project to use ag2 now, autogen is separated into microsoft and ag2

Browse Source
This commit is contained in:
hackertron
2025-03-05 18:54:16 +08:00
parent f69f43550a
commit f9da7a5f83
6 changed files with 51 additions and 500 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
dkim_skill.py
View File

@@ -1,32 +0,0 @@
import dkim
import email
import sys
def verify_dkim_signature(email_file):
try:
# Open and read the email file
with open(email_file, 'rb') as f:
raw_email = f.read()
# Parse the email
parsed_email = email.message_from_bytes(raw_email)
# Extract the headers and body
headers = parsed_email.as_bytes().split(b'\r\n\r\n', 1)[0]
body = parsed_email.get_payload(decode=True)
# Verify DKIM signature
if dkim.verify(raw_email):
print("DKIM signature is valid.")
else:
print("DKIM signature is invalid.")
except Exception as e:
print(f"An error occurred: {e}")
if __name__ == "__main__":
if len(sys.argv) != 2:
print("Usage: python verify_dkim.py <path_to_email_file>")
sys.exit(1)
email_file = sys.argv[1]
verify_dkim_signature(email_file)

53
main.py
View File

@@ -1,26 +1,15 @@
import autogen
import os
import json
from dotenv import load_dotenv
from autogen import AssistantAgent, UserProxyAgent, config_list_from_json
from typing import Annotated
import requests
from system_prompts import front_desk_assistant_prompt, email_assistant_prompt, salary_slip_assistant_prompt
from extract_pdf_skill import process_pdf_from_url
from system_prompts import front_desk_assistant_prompt, email_assistant_prompt, verify_tlsn_proof_prompt
load_dotenv() # take environment variables from .env.
config_list = [
{
'model': 'gpt-3.5-turbo',
'api_key': os.getenv('OPENAI_API_KEY'),
}
]
llm_config = {
"timeout": 120,
"seed": 42, # for caching. once task is run it caches the response,
"config_list": config_list,
"temperature": 0 #lower temperature more standard lesss creative response, higher is more creative
"config_list": config_list_from_json(env_or_file="OAI_CONFIG_LIST.json"),
"temperature": 0
}
def verify_email_with_prove_api(domain :Annotated[str, "The domain name to verify"]) -> Annotated[dict, "The response from the Prove Email API"] | None:
@@ -29,34 +18,31 @@ def verify_email_with_prove_api(domain :Annotated[str, "The domain name to verif
print("response : ", response)
return response.json() if response.status_code == 200 else None
front_desk_assistant = autogen.AssistantAgent(
front_desk_assistant = AssistantAgent(
name="front_desk_assistant",
llm_config=llm_config,
system_message=front_desk_assistant_prompt,
)
email_assistant = autogen.AssistantAgent(
email_assistant = AssistantAgent(
name="email_assistant",
llm_config=llm_config,
system_message=email_assistant_prompt
)
salary_slip_assistant = autogen.AssistantAgent(
name="salary_slip_assistant",
verify_tlsn_proof_assistant = AssistantAgent(
name="verify_tlsn_proof_assistant",
llm_config=llm_config,
system_message=salary_slip_assistant_prompt
system_message=verify_tlsn_proof_prompt
)
# assistant = autogen.AssistantAgent(
# name="laon_assistant",
# salary_slip_assistant = AssistantAgent(
# name="salary_slip_assistant",
# llm_config=llm_config,
# system_message="checks the bank documents. extract pdf using extract_pdf_skill.",
# system_message=salary_slip_assistant_prompt
# )
user_proxy = autogen.UserProxyAgent(
user_proxy = UserProxyAgent(
name="user_proxy",
human_input_mode="ALWAYS",
max_consecutive_auto_reply=3,
@@ -72,18 +58,19 @@ user_proxy = autogen.UserProxyAgent(
)
user_proxy.register_for_llm(name="verify_email_with_prove_api", description="verify email's dkim using prove api verify_email_with_prove_api")(verify_email_with_prove_api)
user_proxy.register_for_execution(name="verify_email_with_prove_api")(verify_email_with_prove_api)
user_proxy.register_for_llm(name="process_pdf_from_url", description="process pdf from url using extract_pdf_skill")(process_pdf_from_url)
user_proxy.register_for_execution(name="process_pdf_from_url")(process_pdf_from_url)
#user_proxy.register_for_llm(name="process_pdf_from_url", description="process pdf from url using extract_pdf_skill")(process_pdf_from_url)
#user_proxy.register_for_execution(name="process_pdf_from_url")(process_pdf_from_url)
def main():
# Register the verify_email_with_prove_api function for the email_assistant
email_assistant.register_function(
function_map={
"verify_email_with_prove_api": verify_email_with_prove_api,
"process_pdf_from_url": process_pdf_from_url
#"process_pdf_from_url": process_pdf_from_url
}
)
chat_results = user_proxy.initiate_chats([
@@ -100,8 +87,8 @@ def main():
"summary_method": "reflection_with_llm"
},
{
"recipient": salary_slip_assistant,
"message": "guide user to upload a salary slip in pdf format and call process_pdf_from_url function to verify the pdf",
"recipient": verify_tlsn_proof_assistant,
"message": "guide user to provide the tlsn proof",
"silent": False,
"summary_method": "reflection_with_llm"
}

86
plugin_agent/plugin_agent_gen.py
View File

@@ -1,86 +0,0 @@
import autogen
import os
import json
from dotenv import load_dotenv
from typing import Annotated
import requests
from plugin_prompts import bank_plugin_prompts
load_dotenv() # take environment variables from .env.
config_list = [
{
'model': 'gpt-3.5-turbo',
'api_key': os.getenv('OPENAI_API_KEY'),
}
]
llm_config = {
"timeout": 120,
"seed": 42, # for caching. once task is run it caches the response,
"config_list": config_list,
"temperature": 0 #lower temperature more standard lesss creative response, higher is more creative
}
def write_plugin(website_url:Annotated[str, "website url"], plugin_name:Annotated[str, "plugin name"]) -> Annotated[dict, "The response from the plugin"] | None:
api_url = f"https://archive.prove.email/api/key?domain={domain}"
response = requests.get(api_url)
print("response : ", response)
return response.json() if response.status_code == 200 else None
plugin_agent = autogen.AssistantAgent(
name="plugin_agent",
llm_config=llm_config,
system_message="""you will write TLSN plugin for the website that the user will provide.
You will ask the user to provide the website url,
API and some web request and response of it and plugin name.
Once you receive the url, use the write_plugin function to write the plugin.
The function will return the response from the plugin.
Analyze the response to gather the following information from the plugin: plugin name, plugin description, plugin author, plugin version, website url.
""",
)
user_proxy = autogen.UserProxyAgent(
name="user_proxy",
human_input_mode="ALWAYS",
max_consecutive_auto_reply=3,
is_termination_msg=lambda x: x.get("content", "").rstrip().endswith("TERMINATE"),
code_execution_config={
"last_n_messages": 3,
"work_dir": "code",
"use_docker": False,
},
llm_config=llm_config,
system_message="""Reply TERMINATE if the task has been solved at full satisfaction
otherwise, reply CONTINUE, or the reason why the task is not solved yet."""
)
user_proxy.register_for_llm(name="write_plugin", description="write TLSN extension plugin for website that user will provide")(write_plugin)
user_proxy.register_for_execution(name="write_plugin")(write_plugin)
def main():
# Register the verify_email_with_prove_api function for the email_assistant
plugin_agent.register_function(
function_map={
"write_plugin": write_plugin,
}
)
chat_results = user_proxy.initiate_chats([
{
"recipient": plugin_agent,
"message": "I want to write a TLSN extension plugin for the website that I will provide, please help me",
"silent": False,
"summary_method": "reflection_with_llm"
}
])
for i, chat_res in enumerate(chat_results):
print(f"*****{i}th chat*******:")
print(chat_res.summary)
print("Human input in the middle:", chat_res.human_input)
print("Conversation cost: ", chat_res.cost)
print("\n\n")
if __name__ == "__main__":
main()

233
plugin_agent/plugin_prompts.py
View File

@@ -1,233 +0,0 @@
bank_plugin_prompts = """
your task is to build tlsn browser extension plugins for the website. I can give you sample code below, that you can refer
to make such plugins. I will give you a task description and you have to build the plugin to do the task.
You will ask users to give the website url, API link / route, auth method.
Here is the task description:
Ask users to provide the website url, API link. Make sure to ask user if TLSN extension is installed. If not, ask them to install it.
Ask users to provide the auth method. Make sure to ask user if TLSN extension is installed. If not, ask them to install it.
Ask users to provide the API link. Make sure to ask user if TLSN extension is installed. If not, ask them to install it.
Here is the sample code below for twitter profile notarization example
import icon from '../assets/icon.png';
import config_json from '../config.json';
import { redirect, notarize, outputJSON, getCookiesByHost, getHeadersByHost } from './utils/hf.js';
/**
* Plugin configuration
* This configurations defines the plugin, most importantly:
* * the different steps
* * the user data (headers, cookies) it will access
* * the web requests it will query (or notarize)
*/
export function config() {
outputJSON({
...config_json,
icon: icon
});
}
function isValidHost(urlString: string) {
const url = new URL(urlString);
return url.hostname === 'twitter.com' || url.hostname === 'x.com';
}
/**
* Implementation of the first (start) plugin step
*/
export function start() {
if (!isValidHost(Config.get('tabUrl'))) {
redirect('https://x.com');
outputJSON(false);
return;
}
outputJSON(true);
}
/**
* Implementation of step "two".
* This step collects and validates authentication cookies and headers for 'api.x.com'.
* If all required information, it creates the request object.
* Note that the url needs to be specified in the `config` too, otherwise the request will be refused.
*/
export function two() {
const cookies = getCookiesByHost('api.x.com');
const headers = getHeadersByHost('api.x.com');
if (
!cookies.auth_token ||
!cookies.ct0 ||
!headers['x-csrf-token'] ||
!headers['authorization']
) {
outputJSON(false);
return;
}
outputJSON({
url: 'https://api.x.com/1.1/account/settings.json',
method: 'GET',
headers: {
'x-twitter-client-language': 'en',
'x-csrf-token': headers['x-csrf-token'],
Host: 'api.x.com',
authorization: headers.authorization,
Cookie: `lang=en; auth_token=${cookies.auth_token}; ct0=${cookies.ct0}`,
'Accept-Encoding': 'identity',
Connection: 'close',
},
secretHeaders: [
`x-csrf-token: ${headers['x-csrf-token']}`,
`cookie: lang=en; auth_token=${cookies.auth_token}; ct0=${cookies.ct0}`,
`authorization: ${headers.authorization}`,
],
});
}
/**
* This method is used to parse the Twitter response and specify what information is revealed (i.e. **not** redacted)
* This method is optional in the notarization request. When it is not specified nothing is redacted.
*
* In this example it locates the `screen_name` and excludes that range from the revealed response.
*/
export function parseTwitterResp() {
const bodyString = Host.inputString();
const params = JSON.parse(bodyString);
if (params.screen_name) {
const revealed = `"screen_name":"${params.screen_name}"`;
const selectionStart = bodyString.indexOf(revealed);
const selectionEnd =
selectionStart + revealed.length;
const secretResps = [
bodyString.substring(0, selectionStart),
bodyString.substring(selectionEnd, bodyString.length),
];
outputJSON(secretResps);
} else {
outputJSON(false);
}
}
/**
* Step 3: calls the `notarize` host function
*/
export function three() {
const params = JSON.parse(Host.inputString());
if (!params) {
outputJSON(false);
} else {
const id = notarize({
...params,
getSecretResponse: 'parseTwitterResp',
});
outputJSON(id);
}
}
Below is the sample config.json that above code uses:
{
"title": "Twitter Profile",
"description": "Notarize ownership of a twitter profile",
"steps": [
{
"title": "Visit Twitter website",
"cta": "Go to x.com",
"action": "start"
},
{
"title": "Collect credentials",
"description": "Login to your account if you haven't already",
"cta": "Check cookies",
"action": "two"
},
{
"title": "Notarize twitter profile",
"cta": "Notarize",
"action": "three",
"prover": true
}
],
"hostFunctions": [
"redirect",
"notarize"
],
"cookies": [
"api.x.com"
],
"headers": [
"api.x.com"
],
"requests": [
{
"url": "https://api.x.com/1.1/account/settings.json",
"method": "GET"
}
]
}
You need to make the config json for user's use case. You can use the sample config.json as a reference.
below is the hf.js file that above code uses:
function redirect(url) {
const { redirect } = Host.getFunctions();
const mem = Memory.fromString(url);
redirect(mem.offset);
}
function notarize(options) {
const { notarize } = Host.getFunctions();
const mem = Memory.fromString(JSON.stringify(options));
const idOffset = notarize(mem.offset);
const id = Memory.find(idOffset).readString();
return id;
}
function outputJSON(json) {
Host.outputString(
JSON.stringify(json),
);
}
function getCookiesByHost(hostname) {
const cookies = JSON.parse(Config.get('cookies'));
if (!cookies[hostname]) throw new Error(`cannot find cookies for ${hostname}`);
return cookies[hostname];
}
function getHeadersByHost(hostname) {
const headers = JSON.parse(Config.get('headers'));
if (!headers[hostname]) throw new Error(`cannot find headers for ${hostname}`);
return headers[hostname];
}
module.exports = {
redirect,
notarize,
outputJSON,
getCookiesByHost,
getHeadersByHost,
};
You have to build the plugin to do the task. You can use the sample code as a reference.
You have to use the following format to build the plugin:
Plugin Name: {plugin_name} // name will be asked by user
Plugin Description: {plugin_description}
Plugin Code: // you will generate plugin code based on the sample code above
{plugin_code}
Remember to use the sample code as a reference and not copy and paste it directly.
Your response should be in the following format:
Plugin Name: {plugin_name}
Plugin Description: {plugin_description}
Plugin Code:
{plugin_code}
"""

145
requirements.txt
View File

@@ -1,124 +1,39 @@
aiohttp==3.9.5
aiosignal==1.3.1
altair==5.3.0
ag2==0.7.6
aiofiles==24.1.0
annotated-types==0.7.0
anyio==4.4.0
attrs==23.2.0
backoff==2.2.1
beautifulsoup4==4.12.3
blinker==1.8.2
cachetools==5.3.3
certifi==2024.6.2
cffi==1.16.0
chardet==5.2.0
charset-normalizer==3.3.2
click==8.1.7
colorama==0.4.6
cryptography==42.0.8
dataclasses-json==0.6.7
deepdiff==7.0.1
anyio==4.8.0
asyncer==0.0.8
autogen-core==0.4.8
certifi==2025.1.31
charset-normalizer==3.4.1
Deprecated==1.2.18
diskcache==5.6.3
distro==1.9.0
dkimpy==1.1.8
dnspython==2.6.1
docker==7.1.0
emoji==2.12.1
filetype==1.2.0
FLAML==2.1.2
frozenlist==1.4.1
gitdb==4.0.11
GitPython==3.1.43
greenlet==3.0.3
fast-depends==2.4.12
h11==0.14.0
httpcore==1.0.5
httpx==0.27.0
idna==3.7
Jinja2==3.1.4
joblib==1.4.2
jsonpatch==1.33
jsonpath-python==1.0.6
jsonpointer==3.0.0
jsonschema==4.22.0
jsonschema-specifications==2023.12.1
langchain==0.2.4
langchain-core==0.2.6
langchain-text-splitters==0.2.1
langdetect==1.0.9
langsmith==0.1.77
lxml==5.2.2
markdown-it-py==3.0.0
MarkupSafe==2.1.5
marshmallow==3.21.3
mdurl==0.1.2
multidict==6.0.5
mypy-extensions==1.0.0
nest-asyncio==1.6.0
nltk==3.8.1
numpy==1.26.4
ollama==0.2.1
openai==1.35.7
opencv-contrib-python==4.10.0.84
opencv-python==4.10.0.84
ordered-set==4.1.0
orjson==3.10.5
packaging==24.1
pandas==2.2.2
pdf2image==1.17.0
pdfminer==20191125
pdfminer.six==20231228
phi==0.6.7
phidata==2.4.20
pillow==10.3.0
pillow_heif==0.16.0
protobuf==4.25.3
pyarrow==16.1.0
pyautogen==0.2.31
pycparser==2.22
pycryptodome==3.20.0
pydantic==2.7.4
pydantic-settings==2.3.3
pydantic_core==2.18.4
pydeck==0.9.1
Pygments==2.18.0
pypdf==4.2.0
PyPDF2==3.0.1
python-dateutil==2.9.0.post0
httpcore==1.0.7
httpx==0.28.1
idna==3.10
importlib_metadata==8.5.0
jiter==0.8.2
jsonref==1.1.0
openai==1.65.3
opentelemetry-api==1.30.0
packaging==24.2
pillow==11.1.0
protobuf==5.29.3
pyautogen==0.7.6
pydantic==2.10.6
pydantic_core==2.27.2
python-dotenv==1.0.1
python-iso639==2024.4.27
python-magic==0.4.27
pytz==2024.1
pywin32==306
PyYAML==6.0.1
rapidfuzz==3.9.3
referencing==0.35.1
regex==2024.5.15
regex==2024.11.6
requests==2.32.3
requests-toolbelt==1.0.0
rich==13.7.1
rpds-py==0.18.1
shellingham==1.5.4
six==1.16.0
smmap==5.0.1
sniffio==1.3.1
soupsieve==2.5
SQLAlchemy==2.0.30
streamlit==1.35.0
tabulate==0.9.0
tenacity==8.3.0
termcolor==2.4.0
tiktoken==0.7.0
toml==0.10.2
tomli==2.0.1
toolz==0.12.1
tornado==6.4.1
tqdm==4.66.4
typer==0.12.3
typing-inspect==0.9.0
termcolor==2.5.0
tiktoken==0.9.0
tqdm==4.67.1
typing_extensions==4.12.2
tzdata==2024.1
unstructured==0.14.6
unstructured-client==0.23.3
urllib3==2.2.1
watchdog==4.0.1
wrapt==1.16.0
yarl==1.9.4
urllib3==2.3.0
wrapt==1.17.2
zipp==3.21.0

2
system_prompts.py
View File

@@ -7,7 +7,7 @@ front_desk_assistant_prompt = """You have a personality of monopoly banker. You
that json response to a file called bank.json
"""
email_assistant_prompt = """You will have access to bank.json from front_desk_assistant.
email_assistant_prompt = """You will have access to bank.json from front_desk_assistant. it's in code folder.
You will guide user to paste their raw email. Assume user has desktop and not on their mobile phone.
guide user to paste their raw email to you. Tell them to paste raw email in chunks, not the complete email in one go.
You will then analyze the email and check if it's valid and details matches with bank.json."""