mirror of
https://github.com/coleam00/Archon.git
synced 2026-01-05 14:18:43 -05:00
The New Archon (Beta) - The Operating System for AI Coding Assistants!
This commit is contained in:
@@ -0,0 +1,34 @@
|
||||
@github_agent.tool
|
||||
async def get_file_content(ctx: RunContext[GitHubDeps], github_url: str, file_path: str) -> str:
|
||||
"""Get the content of a specific file from the GitHub repository.
|
||||
|
||||
Args:
|
||||
ctx: The context.
|
||||
github_url: The GitHub repository URL.
|
||||
file_path: Path to the file within the repository.
|
||||
|
||||
Returns:
|
||||
str: File content as a string.
|
||||
"""
|
||||
match = re.search(r'github\.com[:/]([^/]+)/([^/]+?)(?:\.git)?$', github_url)
|
||||
if not match:
|
||||
return "Invalid GitHub URL format"
|
||||
|
||||
owner, repo = match.groups()
|
||||
headers = {'Authorization': f'token {ctx.deps.github_token}'} if ctx.deps.github_token else {}
|
||||
|
||||
response = await ctx.deps.client.get(
|
||||
f'https://raw.githubusercontent.com/{owner}/{repo}/main/{file_path}',
|
||||
headers=headers
|
||||
)
|
||||
|
||||
if response.status_code != 200:
|
||||
# Try with master branch if main fails
|
||||
response = await ctx.deps.client.get(
|
||||
f'https://raw.githubusercontent.com/{owner}/{repo}/master/{file_path}',
|
||||
headers=headers
|
||||
)
|
||||
if response.status_code != 200:
|
||||
return f"Failed to get file content: {response.text}"
|
||||
|
||||
return response.text
|
||||
@@ -0,0 +1,42 @@
|
||||
@github_agent.tool
|
||||
async def get_repo_structure(ctx: RunContext[GitHubDeps], github_url: str) -> str:
|
||||
"""Get the directory structure of a GitHub repository.
|
||||
|
||||
Args:
|
||||
ctx: The context.
|
||||
github_url: The GitHub repository URL.
|
||||
|
||||
Returns:
|
||||
str: Directory structure as a formatted string.
|
||||
"""
|
||||
match = re.search(r'github\.com[:/]([^/]+)/([^/]+?)(?:\.git)?$', github_url)
|
||||
if not match:
|
||||
return "Invalid GitHub URL format"
|
||||
|
||||
owner, repo = match.groups()
|
||||
headers = {'Authorization': f'token {ctx.deps.github_token}'} if ctx.deps.github_token else {}
|
||||
|
||||
response = await ctx.deps.client.get(
|
||||
f'https://api.github.com/repos/{owner}/{repo}/git/trees/main?recursive=1',
|
||||
headers=headers
|
||||
)
|
||||
|
||||
if response.status_code != 200:
|
||||
# Try with master branch if main fails
|
||||
response = await ctx.deps.client.get(
|
||||
f'https://api.github.com/repos/{owner}/{repo}/git/trees/master?recursive=1',
|
||||
headers=headers
|
||||
)
|
||||
if response.status_code != 200:
|
||||
return f"Failed to get repository structure: {response.text}"
|
||||
|
||||
data = response.json()
|
||||
tree = data['tree']
|
||||
|
||||
# Build directory structure
|
||||
structure = []
|
||||
for item in tree:
|
||||
if not any(excluded in item['path'] for excluded in ['.git/', 'node_modules/', '__pycache__/']):
|
||||
structure.append(f"{'📁 ' if item['type'] == 'tree' else '📄 '}{item['path']}")
|
||||
|
||||
return "\n".join(structure)
|
||||
@@ -0,0 +1,38 @@
|
||||
@github_agent.tool
|
||||
async def get_repo_info(ctx: RunContext[GitHubDeps], github_url: str) -> str:
|
||||
"""Get repository information including size and description using GitHub API.
|
||||
|
||||
Args:
|
||||
ctx: The context.
|
||||
github_url: The GitHub repository URL.
|
||||
|
||||
Returns:
|
||||
str: Repository information as a formatted string.
|
||||
"""
|
||||
match = re.search(r'github\.com[:/]([^/]+)/([^/]+?)(?:\.git)?$', github_url)
|
||||
if not match:
|
||||
return "Invalid GitHub URL format"
|
||||
|
||||
owner, repo = match.groups()
|
||||
headers = {'Authorization': f'token {ctx.deps.github_token}'} if ctx.deps.github_token else {}
|
||||
|
||||
response = await ctx.deps.client.get(
|
||||
f'https://api.github.com/repos/{owner}/{repo}',
|
||||
headers=headers
|
||||
)
|
||||
|
||||
if response.status_code != 200:
|
||||
return f"Failed to get repository info: {response.text}"
|
||||
|
||||
data = response.json()
|
||||
size_mb = data['size'] / 1024
|
||||
|
||||
return (
|
||||
f"Repository: {data['full_name']}\n"
|
||||
f"Description: {data['description']}\n"
|
||||
f"Size: {size_mb:.1f}MB\n"
|
||||
f"Stars: {data['stargazers_count']}\n"
|
||||
f"Language: {data['language']}\n"
|
||||
f"Created: {data['created_at']}\n"
|
||||
f"Last Updated: {data['updated_at']}"
|
||||
)
|
||||
@@ -0,0 +1,48 @@
|
||||
@web_search_agent.tool
|
||||
async def search_web(
|
||||
ctx: RunContext[Deps], web_query: str
|
||||
) -> str:
|
||||
"""Search the web given a query defined to answer the user's question.
|
||||
|
||||
Args:
|
||||
ctx: The context.
|
||||
web_query: The query for the web search.
|
||||
|
||||
Returns:
|
||||
str: The search results as a formatted string.
|
||||
"""
|
||||
if ctx.deps.brave_api_key is None:
|
||||
return "This is a test web search result. Please provide a Brave API key to get real search results."
|
||||
|
||||
headers = {
|
||||
'X-Subscription-Token': ctx.deps.brave_api_key,
|
||||
'Accept': 'application/json',
|
||||
}
|
||||
|
||||
with logfire.span('calling Brave search API', query=web_query) as span:
|
||||
r = await ctx.deps.client.get(
|
||||
'https://api.search.brave.com/res/v1/web/search',
|
||||
params={
|
||||
'q': web_query,
|
||||
'count': 5,
|
||||
'text_decorations': True,
|
||||
'search_lang': 'en'
|
||||
},
|
||||
headers=headers
|
||||
)
|
||||
r.raise_for_status()
|
||||
data = r.json()
|
||||
span.set_attribute('response', data)
|
||||
|
||||
results = []
|
||||
|
||||
# Add web results in a nice formatted way
|
||||
web_results = data.get('web', {}).get('results', [])
|
||||
for item in web_results[:3]:
|
||||
title = item.get('title', '')
|
||||
description = item.get('description', '')
|
||||
url = item.get('url', '')
|
||||
if title and description:
|
||||
results.append(f"Title: {title}\nSummary: {description}\nSource: {url}\n")
|
||||
|
||||
return "\n".join(results) if results else "No results found for the query."
|
||||
Reference in New Issue
Block a user