Files
fil/docs/snippets/zig/advanced/keyword_extraction_example.md
Henrik Jess Nielsen b4c07d3693
All checks were successful
Deploy fil (kreuzberg) / deploy (push) Successful in 49s
Nomad changes
2026-06-01 23:40:55 +02:00

1.5 KiB

const std = @import("std");
const kreuzberg = @import("kreuzberg");

pub fn main() !void {
    var gpa = std.heap.GeneralPurposeAllocator(.{}){};
    defer _ = gpa.deinit();
    const allocator = gpa.allocator();

    const config_json =
        \\{
        \\  "keywords": {
        \\    "algorithm": "yake",
        \\    "max_keywords": 10,
        \\    "min_score": 0.3,
        \\    "ngram_range": [1, 3]
        \\  }
        \\}
    ;

    const result_json = try kreuzberg.extract_file_sync("research_paper.pdf", null, config_json);
    defer std.heap.c_allocator.free(result_json);

    var parsed = try std.json.parseFromSlice(std.json.Value, allocator, result_json, .{});
    defer parsed.deinit();

    const root = parsed.value;
    if (root != .object) return;

    const stdout = std.io.getStdOut().writer();

    const keywords_val = root.object.get("extracted_keywords") orelse return;
    if (keywords_val != .array) return;

    for (keywords_val.array.items) |keyword| {
        if (keyword != .object) continue;

        const text_val = keyword.object.get("text") orelse continue;
        const score_val = keyword.object.get("score") orelse continue;
        if (text_val != .string) continue;

        const score: f64 = switch (score_val) {
            .float => |f| f,
            .integer => |i| @floatFromInt(i),
            else => continue,
        };

        try stdout.print("{s}: {d:.4}\n", .{ text_val.string, score });
    }
}