react-native-gemma-agent 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +21 -0
- package/README.md +457 -0
- package/package.json +52 -0
- package/skills/calculator.ts +47 -0
- package/skills/deviceLocation.ts +180 -0
- package/skills/index.ts +3 -0
- package/skills/queryWikipedia.ts +96 -0
- package/skills/readCalendar.ts +74 -0
- package/skills/webSearch.ts +75 -0
- package/src/AgentOrchestrator.ts +315 -0
- package/src/BM25Scorer.ts +118 -0
- package/src/FunctionCallParser.ts +113 -0
- package/src/GemmaAgentProvider.tsx +101 -0
- package/src/InferenceEngine.ts +301 -0
- package/src/ModelManager.ts +244 -0
- package/src/SkillRegistry.ts +60 -0
- package/src/SkillSandbox.tsx +155 -0
- package/src/index.ts +52 -0
- package/src/types.ts +197 -0
- package/src/useGemmaAgent.ts +222 -0
- package/src/useModelDownload.ts +80 -0
- package/src/useSkillRegistry.ts +58 -0
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { useState, useCallback } from 'react';
|
|
2
|
+
import { useGemmaAgentContext } from './GemmaAgentProvider';
|
|
3
|
+
import type { SkillManifest } from './types';
|
|
4
|
+
|
|
5
|
+
export type UseSkillRegistryReturn = {
|
|
6
|
+
/** Register a new skill */
|
|
7
|
+
registerSkill: (skill: SkillManifest) => void;
|
|
8
|
+
/** Remove a skill by name */
|
|
9
|
+
unregisterSkill: (name: string) => void;
|
|
10
|
+
/** All registered skills */
|
|
11
|
+
skills: SkillManifest[];
|
|
12
|
+
/** Check if a skill is registered */
|
|
13
|
+
hasSkill: (name: string) => boolean;
|
|
14
|
+
/** Remove all skills */
|
|
15
|
+
clear: () => void;
|
|
16
|
+
};
|
|
17
|
+
|
|
18
|
+
export function useSkillRegistry(): UseSkillRegistryReturn {
|
|
19
|
+
const { registry } = useGemmaAgentContext();
|
|
20
|
+
|
|
21
|
+
// Counter to trigger re-renders when skills change
|
|
22
|
+
const [, setVersion] = useState(0);
|
|
23
|
+
const bump = () => setVersion((v) => v + 1);
|
|
24
|
+
|
|
25
|
+
const registerSkill = useCallback(
|
|
26
|
+
(skill: SkillManifest) => {
|
|
27
|
+
registry.registerSkill(skill);
|
|
28
|
+
bump();
|
|
29
|
+
},
|
|
30
|
+
[registry],
|
|
31
|
+
);
|
|
32
|
+
|
|
33
|
+
const unregisterSkill = useCallback(
|
|
34
|
+
(name: string) => {
|
|
35
|
+
registry.unregisterSkill(name);
|
|
36
|
+
bump();
|
|
37
|
+
},
|
|
38
|
+
[registry],
|
|
39
|
+
);
|
|
40
|
+
|
|
41
|
+
const clear = useCallback(() => {
|
|
42
|
+
registry.clear();
|
|
43
|
+
bump();
|
|
44
|
+
}, [registry]);
|
|
45
|
+
|
|
46
|
+
const hasSkill = useCallback(
|
|
47
|
+
(name: string) => registry.hasSkill(name),
|
|
48
|
+
[registry],
|
|
49
|
+
);
|
|
50
|
+
|
|
51
|
+
return {
|
|
52
|
+
registerSkill,
|
|
53
|
+
unregisterSkill,
|
|
54
|
+
skills: registry.getSkills(),
|
|
55
|
+
hasSkill,
|
|
56
|
+
clear,
|
|
57
|
+
};
|
|
58
|
+
}
|