File size: 562 Bytes
57ae862 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | #ifndef NEEDLE_H
#define NEEDLE_H
#ifndef NEEDLE_API
#define NEEDLE_API __attribute__((visibility("default")))
#endif
#ifdef __cplusplus
extern "C" {
#endif
NEEDLE_API int needle_init(
const char* system_prompt,
const char* tools_json,
const char* tool_index_path
);
NEEDLE_API int needle_complete(
const char* input,
int max_new_tokens,
char* out,
int out_capacity
);
NEEDLE_API void needle_reset(void);
NEEDLE_API int needle_load(
const unsigned char* cact,
unsigned long long n
);
#ifdef __cplusplus
}
#endif
#endif
|