@llmindset/hf-mcp 0.1.16

package/test/fixtures/space-result.json

@@ -0,0 +1,263 @@
+[
+	{
+		"author": "ginipick",
+		"authorData": {
+			"_id": "65acd60b57f263e3d0ff0647",
+			"avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/65acd60b57f263e3d0ff0647/HENR9sR3CDchSDldrYOdS.png",
+			"fullname": "ginipick",
+			"name": "ginipick",
+			"type": "user",
+			"isPro": true,
+			"isHf": false,
+			"isHfAdmin": false,
+			"isMod": false,
+			"followerCount": 602
+		},
+		"colorFrom": "gray",
+		"colorTo": "pink",
+		"createdAt": "2024-08-22T02:55:22.000Z",
+		"emoji": "🦀🏆🦀",
+		"id": "ginipick/FLUXllama",
+		"lastModified": "2025-05-21T13:20:06.000Z",
+		"likes": 378,
+		"pinned": false,
+		"private": false,
+		"sdk": "gradio",
+		"repoType": "space",
+		"runtime": {
+			"stage": "RUNNING",
+			"hardware": {
+				"current": "zero-a10g",
+				"requested": "zero-a10g"
+			},
+			"storage": null,
+			"gcTimeout": 172800,
+			"replicas": {
+				"current": 1,
+				"requested": 1
+			},
+			"devMode": false,
+			"domains": [
+				{
+					"domain": "ginipick-fluxllama.hf.space",
+					"stage": "READY"
+				}
+			],
+			"sha": "36a649401e539c622e2a397528ad02ce5f84fe20"
+		},
+		"shortDescription": "mcp_server & FLUX 4-bit Quantization(just 8GB VRAM)",
+		"title": "FLUXllama",
+		"isLikedByUser": false,
+		"originSpace": {
+			"name": "nyanko7/flux1-dev-nf4",
+			"author": {
+				"_id": "63467f5a7fb9f11870c411ee",
+				"avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/63467f5a7fb9f11870c411ee/Ztk8IModHk8lDxZWnM1DJ.png",
+				"fullname": "Nyanko",
+				"name": "nyanko7",
+				"type": "user",
+				"isPro": false,
+				"isHf": false,
+				"isHfAdmin": false,
+				"isMod": false,
+				"followerCount": 206
+			}
+		},
+		"ai_short_description": "Generate images using text prompts",
+		"ai_category": "Image Generation",
+		"trendingScore": 8,
+		"semanticRelevancyScore": 0.9137691233545246
+	},
+	{
+		"author": "ginipick",
+		"authorData": {
+			"_id": "65acd60b57f263e3d0ff0647",
+			"avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/65acd60b57f263e3d0ff0647/HENR9sR3CDchSDldrYOdS.png",
+			"fullname": "ginipick",
+			"name": "ginipick",
+			"type": "user",
+			"isPro": true,
+			"isHf": false,
+			"isHfAdmin": false,
+			"isMod": false,
+			"followerCount": 602
+		},
+		"colorFrom": "yellow",
+		"colorTo": "pink",
+		"createdAt": "2024-09-14T01:07:20.000Z",
+		"emoji": "💬⚡",
+		"id": "ginipick/Realtime-FLUX",
+		"lastModified": "2025-05-21T13:28:38.000Z",
+		"likes": 106,
+		"pinned": true,
+		"private": false,
+		"sdk": "gradio",
+		"repoType": "space",
+		"runtime": {
+			"stage": "RUNNING",
+			"hardware": {
+				"current": "zero-a10g",
+				"requested": "zero-a10g"
+			},
+			"storage": null,
+			"gcTimeout": 172800,
+			"replicas": {
+				"current": 1,
+				"requested": 1
+			},
+			"devMode": false,
+			"domains": [
+				{
+					"domain": "ginipick-realtime-flux.hf.space",
+					"stage": "READY"
+				}
+			],
+			"sha": "6293da23835f10d329d91eb82fb32ba90f7b7b6d"
+		},
+		"shortDescription": "mcp_server & High quality Images in Realtime",
+		"title": "Realtime FLUX Image",
+		"isLikedByUser": false,
+		"originSpace": {
+			"name": "KingNish/Realtime-FLUX",
+			"author": {
+				"_id": "6612aedf09f16e7347dfa7e1",
+				"avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/6612aedf09f16e7347dfa7e1/bPYjBXCedY_1fSIPjoBTY.jpeg",
+				"fullname": "Nishith Jain",
+				"name": "KingNish",
+				"type": "user",
+				"isPro": false,
+				"isHf": false,
+				"isHfAdmin": false,
+				"isMod": false,
+				"followerCount": 1206
+			}
+		},
+		"ai_short_description": "Generate images from text prompts",
+		"ai_category": "Image Generation",
+		"trendingScore": 3.5,
+		"semanticRelevancyScore": 0.6803295182592884
+	},
+	{
+		"author": "evalstate",
+		"authorData": {
+			"_id": "6319b36409baf858241f0f89",
+			"avatarUrl": "/avatars/909635453bf62a2a7118a01dd51b811c.svg",
+			"fullname": "shaun smith",
+			"name": "evalstate",
+			"type": "user",
+			"isPro": true,
+			"isHf": false,
+			"isHfAdmin": false,
+			"isMod": false,
+			"followerCount": 3
+		},
+		"colorFrom": "yellow",
+		"colorTo": "pink",
+		"createdAt": "2024-12-08T17:01:13.000Z",
+		"emoji": "🏎️💨",
+		"id": "evalstate/flux1_schnell",
+		"lastModified": "2025-04-30T18:45:28.000Z",
+		"likes": 1,
+		"pinned": false,
+		"private": false,
+		"sdk": "gradio",
+		"repoType": "space",
+		"runtime": {
+			"stage": "RUNNING",
+			"hardware": {
+				"current": "zero-a10g",
+				"requested": "zero-a10g"
+			},
+			"storage": null,
+			"gcTimeout": 172800,
+			"replicas": {
+				"current": 1,
+				"requested": 1
+			},
+			"devMode": false,
+			"domains": [
+				{
+					"domain": "evalstate-flux-1-schnell.hf.space",
+					"stage": "READY"
+				},
+				{
+					"domain": "evalstate-flux1-schnell.hf.space",
+					"stage": "READY"
+				}
+			],
+			"sha": "ae85ed512e3324ffbabb8c1e2a2e490fd5e1e252"
+		},
+		"title": "FLUX.1 [Schnell]",
+		"isLikedByUser": false,
+		"originSpace": {
+			"name": "black-forest-labs/FLUX.1-schnell",
+			"author": {
+				"avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/62cfefa74b3e8dc1e32e38bf/GgkglHn3sIo6C5XGTtZSs.png",
+				"fullname": "Black Forest Labs",
+				"name": "black-forest-labs",
+				"type": "org",
+				"isHf": false,
+				"isHfAdmin": false,
+				"isMod": false,
+				"isEnterprise": false,
+				"followerCount": 18355
+			}
+		},
+		"ai_short_description": "Generate images from text prompts",
+		"ai_category": "Image Generation",
+		"trendingScore": 0,
+		"semanticRelevancyScore": 0.25
+	},
+	{
+		"author": "black-forest-labs",
+		"authorData": {
+			"avatarUrl": "https://cdn-avatars.huggingface.co/v1/production/uploads/62cfefa74b3e8dc1e32e38bf/GgkglHn3sIo6C5XGTtZSs.png",
+			"fullname": "Black Forest Labs",
+			"name": "black-forest-labs",
+			"type": "org",
+			"isHf": false,
+			"isHfAdmin": false,
+			"isMod": false,
+			"isEnterprise": false,
+			"followerCount": 18356
+		},
+		"colorFrom": "yellow",
+		"colorTo": "pink",
+		"createdAt": "2024-08-01T09:32:41.000Z",
+		"emoji": "🏎️💨",
+		"id": "black-forest-labs/FLUX.1-schnell",
+		"lastModified": "2024-08-09T14:51:04.000Z",
+		"likes": 4804,
+		"pinned": false,
+		"private": false,
+		"sdk": "gradio",
+		"repoType": "space",
+		"runtime": {
+			"stage": "RUNNING",
+			"hardware": {
+				"current": "zero-a10g",
+				"requested": "zero-a10g"
+			},
+			"storage": null,
+			"gcTimeout": 172800,
+			"replicas": {
+				"current": 1,
+				"requested": "auto"
+			},
+			"devMode": false,
+			"domains": [
+				{
+					"domain": "black-forest-labs-flux-1-schnell.hf.space",
+					"stage": "READY"
+				}
+			],
+			"sha": "86f96a46582e9bbbc6245e4b1eb7d72cd8ec7bda"
+		},
+		"title": "FLUX.1 [Schnell]",
+		"isLikedByUser": false,
+		"ai_short_description": "Generate images from text prompts",
+		"ai_category": "Image Generation",
+		"trendingScore": 24,
+		"semanticRelevancyScore": 0.6919353880044354
+	}
+]

package/test/paper-search.spec.ts

@@ -0,0 +1,57 @@
+import { describe, beforeEach, afterEach, it, expect } from 'vitest';
+import type { PaperSearchResult } from '../src/paper-search.js';
+import { authors } from '../src/paper-search.js';
+import { published } from '../src/paper-search.js';
+import { readFileSync } from 'fs';
+import path from 'path';
+import { formatDate } from '../src/utilities.js';
+
+describe('PaperSearchService', () => {
+	let kazakh: PaperSearchResult[];
+	function loadTestData(filename: string) {
+		const filePath = path.join(__dirname, '../test/fixtures', filename);
+		const fileContent = readFileSync(filePath, 'utf-8');
+		return JSON.parse(fileContent) as PaperSearchResult[];
+	}
+
+	beforeEach(() => {
+		kazakh = loadTestData('paper_result_kazakh.json');
+	});
+
+	afterEach(() => {});
+
+	it('format the published on date correctly', () => {
+		expect(published(kazakh[0]?.paper.publishedAt)).toBe('Published on 6 Apr, 2024');
+	});
+
+	//github.com/huggingface/huggingface_hub/blob/a26b93e8ba0b51ce76ce5c2044896587c47c6b60/src/huggingface_hub/utils/_datetime.py#L50-L62
+	it('handles times with no decimal point', () => {
+		expect(published('2021-08-30T00:04:01Z')).toBe('Published on 30 Aug, 2021');
+	});
+
+	it('deals with bad inputs', () => {
+		expect(published('invalid_date')).toBe('Publication date not available');
+	});
+
+	it('shows us not available for empty list', () => {
+		expect(authors([])).toBe('**Authors:** Not available');
+	});
+
+	it('shows us hf profiles for authors when known ', () => {
+		expect(authors(kazakh[2]?.paper?.authors || [])).toBe(
+			'**Authors:** Rustem Yeshpanov ([yeshpanovrustem](https://hf.co/yeshpanovrustem)), Huseyin Atakan Varol'
+		);
+	});
+
+	it('shows us hf profiles when known ', () => {
+		expect(authors(kazakh[2]?.paper?.authors || [])).toBe(
+			'**Authors:** Rustem Yeshpanov ([yeshpanovrustem](https://hf.co/yeshpanovrustem)), Huseyin Atakan Varol'
+		);
+	});
+
+	it("it won't break with herd of llamas that has 500 authors", () => {
+		expect(authors(kazakh[0]?.paper?.authors || [], 1)).toBe(
+			'**Authors:** Rustem Yeshpanov ([yeshpanovrustem](https://hf.co/yeshpanovrustem)), and 4 more.'
+		);
+	});
+});

package/test/paper-summary.spec.ts

@@ -0,0 +1,113 @@
+import { describe, it, expect } from 'vitest';
+import { extractArxivIdFromInput } from '../src/paper-summary.js';
+
+describe('extractArxivIdFromInput', () => {
+	it('should handle plain arXiv ID format', () => {
+		expect(extractArxivIdFromInput('2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('2301.12345')).toBe('2301.12345');
+		expect(extractArxivIdFromInput('1907.11692')).toBe('1907.11692');
+	});
+
+	it('should handle arxiv: prefix', () => {
+		expect(extractArxivIdFromInput('arxiv:2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('ARXIV:2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('ArXiv:2502.16161')).toBe('2502.16161');
+	});
+
+	it('should handle arxiv. prefix (typo)', () => {
+		expect(extractArxivIdFromInput('arxiv.2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('ARXIV.2502.16161')).toBe('2502.16161');
+	});
+
+	it('should handle Hugging Face paper URLs', () => {
+		expect(extractArxivIdFromInput('https://huggingface.co/papers/2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('https://hf.co/papers/2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('huggingface.co/papers/2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('hf.co/papers/2502.16161')).toBe('2502.16161');
+	});
+
+	it('should handle arXiv.org URLs', () => {
+		expect(extractArxivIdFromInput('https://arxiv.org/abs/2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('arxiv.org/abs/2502.16161')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('http://www.arxiv.org/abs/2502.16161')).toBe('2502.16161');
+	});
+
+	it('should trim whitespace', () => {
+		expect(extractArxivIdFromInput('  2502.16161  ')).toBe('2502.16161');
+		expect(extractArxivIdFromInput('\tarxiv:2502.16161\n')).toBe('2502.16161');
+	});
+
+	it('should reject invalid formats', () => {
+		// Invalid arXiv ID patterns
+		expect(() => extractArxivIdFromInput('2502.161')).toThrow('Invalid arXiv ID format');
+		expect(() => extractArxivIdFromInput('502.16161')).toThrow('Invalid arXiv ID format');
+		expect(() => extractArxivIdFromInput('2502.1616161')).toThrow('Invalid arXiv ID format');
+		expect(() => extractArxivIdFromInput('abcd.efgh')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: abcd.efgh'
+		);
+
+		// Invalid prefixes
+		expect(() => extractArxivIdFromInput('arxiv:2502.161')).toThrow('Invalid arXiv ID format after "arxiv:" prefix');
+		expect(() => extractArxivIdFromInput('arxiv.abcd.efgh')).toThrow('Invalid arXiv ID format after "arxiv." prefix');
+
+		// Invalid domains
+		expect(() => extractArxivIdFromInput('https://example.com/papers/2502.16161')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: example.com'
+		);
+		expect(() => extractArxivIdFromInput('github.com/papers/2502.16161')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: github.com'
+		);
+
+		// Invalid paths on valid domains
+		expect(() => extractArxivIdFromInput('https://arxiv.org/pdf/2502.16161')).toThrow(
+			'arXiv URL must be in format: arxiv.org/abs/YYMM.NNNNN'
+		);
+		expect(() => extractArxivIdFromInput('https://hf.co/models/2502.16161')).toThrow(
+			'Hugging Face URL must be in format: hf.co/papers/YYMM.NNNNN'
+		);
+
+		// Too short
+		expect(() => extractArxivIdFromInput('25')).toThrow('Invalid arXiv ID format');
+		expect(() => extractArxivIdFromInput('')).toThrow('Paper ID is required');
+
+		// With query params or fragments
+		expect(() => extractArxivIdFromInput('hf.co/papers/2502.16161?foo=bar')).toThrow(
+			'URL must contain only the paper ID path'
+		);
+		expect(() => extractArxivIdFromInput('hf.co/papers/2502.16161#section')).toThrow(
+			'URL must contain only the paper ID path'
+		);
+	});
+
+	it('should reject obvious domain names without path', () => {
+		expect(() => extractArxivIdFromInput('hf.co')).toThrow(
+			'Hugging Face URL must be in format: hf.co/papers/YYMM.NNNNN'
+		);
+		expect(() => extractArxivIdFromInput('huggingface.com')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: huggingface.com'
+		);
+		expect(() => extractArxivIdFromInput('arxiv.org')).toThrow('arXiv URL must be in format: arxiv.org/abs/YYMM.NNNNN');
+	});
+
+	it('should handle edge cases for domain validation', () => {
+		// Valid domains with valid paths
+		expect(extractArxivIdFromInput('http://arxiv.org/abs/2301.12345')).toBe('2301.12345');
+		expect(extractArxivIdFromInput('https://www.arxiv.org/abs/2301.12345')).toBe('2301.12345');
+
+		// Invalid domains that might look similar
+		expect(() => extractArxivIdFromInput('arxiv.com/abs/2502.16161')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: arxiv.com'
+		);
+		expect(() => extractArxivIdFromInput('huggingface.ai/papers/2502.16161')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: huggingface.ai'
+		);
+		expect(() => extractArxivIdFromInput('hf.ai/papers/2502.16161')).toThrow(
+			'URL must be from arxiv.org, huggingface.co, or hf.co. Got: hf.ai'
+		);
+
+		// URLs without protocol should also work
+		expect(extractArxivIdFromInput('arxiv.org/abs/2301.12345')).toBe('2301.12345');
+		expect(extractArxivIdFromInput('hf.co/papers/2301.12345')).toBe('2301.12345');
+		expect(extractArxivIdFromInput('huggingface.co/papers/2301.12345')).toBe('2301.12345');
+	});
+});

package/test/space-files.spec.ts

@@ -0,0 +1,232 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { SpaceFilesTool, SPACE_FILES_TOOL_CONFIG } from '../src/space-files.js';
+import * as hub from '@huggingface/hub';
+
+// Mock the @huggingface/hub module
+vi.mock('@huggingface/hub', () => ({
+	spaceInfo: vi.fn(),
+	listFiles: vi.fn(),
+}));
+
+describe('SpaceFilesTool', () => {
+	let tool: SpaceFilesTool;
+
+	beforeEach(() => {
+		tool = new SpaceFilesTool(undefined, 'testuser');
+		vi.clearAllMocks();
+	});
+
+	describe('getSpaceFilesWithUrls', () => {
+		it('should list files for a static space with subdomain', async () => {
+			// Mock space info response
+			vi.mocked(hub.spaceInfo).mockResolvedValue({
+				id: 'evalstate/filedrop',
+				sdk: 'static',
+				subdomain: 'evalstate-filedrop',
+			} as any);
+
+			// Mock file listing
+			const mockFiles = [
+				{
+					type: 'file',
+					path: 'index.html',
+					size: 1024,
+					lastCommit: { date: '2024-01-01T00:00:00Z' },
+					lfs: null,
+				},
+				{
+					type: 'file',
+					path: 'css/style.css',
+					size: 2048,
+					lastCommit: { date: '2024-01-02T00:00:00Z' },
+					lfs: null,
+				},
+			];
+
+			// Create async generator for listFiles
+			async function* mockListFiles() {
+				for (const file of mockFiles) {
+					yield file;
+				}
+			}
+
+			vi.mocked(hub.listFiles).mockReturnValue(mockListFiles() as any);
+
+			const files = await tool.getSpaceFilesWithUrls('evalstate/filedrop');
+
+			expect(files).toHaveLength(2);
+			expect(files[0]).toMatchObject({
+				path: 'css/style.css',
+				size: 2048,
+				type: 'file',
+				url: 'https://huggingface.co/spaces/evalstate/filedrop/resolve/main/css/style.css',
+				sizeFormatted: '2.0 KB',
+				lfs: false,
+			});
+			expect(files[1]).toMatchObject({
+				path: 'index.html',
+				size: 1024,
+				type: 'file',
+				url: 'https://huggingface.co/spaces/evalstate/filedrop/resolve/main/index.html',
+				sizeFormatted: '1.0 KB',
+				lfs: false,
+			});
+		});
+
+		it('should handle spaces without subdomain', async () => {
+			// Mock space info response without subdomain
+			vi.mocked(hub.spaceInfo).mockResolvedValue({
+				id: 'test/space',
+				sdk: 'static',
+			} as any);
+
+			// Mock file listing
+			async function* mockListFiles() {
+				yield {
+					type: 'file',
+					path: 'index.html',
+					size: 512,
+					lastCommit: null,
+					lfs: null,
+				};
+			}
+
+			vi.mocked(hub.listFiles).mockReturnValue(mockListFiles() as any);
+
+			const files = await tool.getSpaceFilesWithUrls('test/space');
+
+			expect(files).toHaveLength(1);
+			expect(files[0].url).toBe('https://huggingface.co/spaces/test/space/resolve/main/index.html');
+		});
+
+		it('should throw error for non-static spaces', async () => {
+			// Mock space info response with non-static SDK
+			vi.mocked(hub.spaceInfo).mockResolvedValue({
+				id: 'test/gradio-app',
+				sdk: 'gradio',
+			} as any);
+
+			await expect(tool.getSpaceFilesWithUrls('test/gradio-app')).rejects.toThrow(
+				'Space "test/gradio-app" is not a static space (found: gradio). This tool only works with static spaces.'
+			);
+		});
+	});
+
+	describe('listFiles', () => {
+		beforeEach(() => {
+			// Mock a successful static space
+			vi.mocked(hub.spaceInfo).mockResolvedValue({
+				id: 'evalstate/filedrop',
+				sdk: 'static',
+				subdomain: 'evalstate-filedrop',
+			} as any);
+
+			// Mock file listing
+			async function* mockListFiles() {
+				yield {
+					type: 'file',
+					path: 'index.html',
+					size: 1024,
+					lastCommit: { date: '2024-01-01T00:00:00Z' },
+					lfs: null,
+				};
+				yield {
+					type: 'file',
+					path: 'js/app.js',
+					size: 4096,
+					lastCommit: { date: '2024-01-02T00:00:00Z' },
+					lfs: null,
+				};
+			}
+
+			vi.mocked(hub.listFiles).mockReturnValue(mockListFiles() as any);
+		});
+
+		it('should generate detailed markdown', async () => {
+			const result = await tool.listFiles({ spaceName: 'evalstate/filedrop' });
+
+			expect(result).toContain('# Files in Space: evalstate/filedrop');
+			expect(result).toContain('**Total Files**: 2');
+			expect(result).toContain('**Total Size**: 5.1 KB');
+			expect(result).toContain('| File Path | Size | Type | Last Modified | URL |');
+			expect(result).toContain('🌐 index.html');
+			expect(result).toContain('📜 app.js');
+			expect(result).toContain('## Direct Access Examples');
+		});
+
+		it('should filter by file type when specified', async () => {
+			const result = await tool.listFiles({ spaceName: 'evalstate/filedrop', fileType: 'all' });
+
+			expect(result).toContain('# Files in Space: evalstate/filedrop');
+			expect(result).toContain('| File Path | Size | Type | Last Modified | URL |');
+			expect(result).toContain('🌐 index.html');
+			expect(result).toContain('📜 app.js');
+			expect(result).toContain('## Direct Access Examples');
+		});
+	});
+
+	describe('SPACE_FILES_TOOL_CONFIG', () => {
+		it('should have correct base configuration', () => {
+			expect(SPACE_FILES_TOOL_CONFIG.name).toBe('space_files');
+			expect(SPACE_FILES_TOOL_CONFIG.annotations.readOnlyHint).toBe(true);
+			expect(SPACE_FILES_TOOL_CONFIG.annotations.destructiveHint).toBe(false);
+		});
+
+		it('should validate schema correctly', () => {
+			const schema = SPACE_FILES_TOOL_CONFIG.schema;
+
+			// Test default values
+			const defaultResult = schema.parse({});
+			expect(defaultResult).toEqual({
+				fileType: 'all',
+			});
+
+			// Test custom values
+			const customResult = schema.parse({
+				spaceName: 'user/space',
+				fileType: 'image',
+			});
+			expect(customResult).toEqual({
+				spaceName: 'user/space',
+				fileType: 'image',
+			});
+
+			// Test invalid fileType
+			expect(() => schema.parse({ fileType: 'invalid' })).toThrow();
+		});
+	});
+
+	describe('createToolConfig', () => {
+		it('should create config with username', () => {
+			const config = SpaceFilesTool.createToolConfig('testuser');
+			expect(config.name).toBe('space_files');
+			expect(config.description).toContain('testuser/filedrop');
+		});
+	});
+
+	describe('default spaceName behavior', () => {
+		it('should use username/filedrop as default when no spaceName provided', async () => {
+			// Mock space info response
+			vi.mocked(hub.spaceInfo).mockResolvedValue({
+				id: 'testuser/filedrop',
+				sdk: 'static',
+				subdomain: 'testuser-filedrop',
+			} as any);
+
+			// Mock empty file listing
+			async function* mockListFiles() {
+				// Empty generator
+			}
+			vi.mocked(hub.listFiles).mockReturnValue(mockListFiles() as any);
+
+			const result = await tool.listFiles({});
+
+			expect(result).toContain('# Files in Space: testuser/filedrop');
+			expect(vi.mocked(hub.spaceInfo)).toHaveBeenCalledWith(
+				expect.objectContaining({
+					name: 'testuser/filedrop',
+				})
+			);
+		});
+	});
+});

package/test/space-search.spec.ts

@@ -0,0 +1,29 @@
+import { describe, beforeEach, afterEach, it, expect } from 'vitest';
+import { readFileSync } from 'fs';
+import path from 'path';
+import { SpaceSearchResult } from '../dist/space-search.js';
+
+describe('SpaceSearchService', () => {
+	let space: SpaceSearchResult[];
+	function loadTestData(filename: string) {
+		const filePath = path.join(__dirname, '../test/fixtures', filename);
+		const fileContent = readFileSync(filePath, 'utf-8');
+		return JSON.parse(fileContent) as SpaceSearchResult[];
+	}
+
+	beforeEach(() => {
+		space = loadTestData('space-result.json');
+	});
+
+	afterEach(() => {});
+
+	it('read the test file', () => {
+		expect('evalstate').toBe(space[2].author);
+	});
+
+	it('picked up other results', () => {
+		expect('RUNNING').toBe(space[2].runtime.stage);
+		expect('Image Generation').toBe(space[2].ai_category);
+		expect('Generate images from text prompts').toBe(space[2].ai_short_description);
+	});
+});