refactor(Capability): Improve code execution efficiency

Author: lvluoyue

src/Capability/PromptChain.php

@@ -16,28 +16,33 @@
  */
 class PromptChain implements PromptGetterInterface, CollectionInterface
 {
-    public function __construct(
-        /**
-         * @var IdentifierInterface[]
-         */
-        private readonly array $items,
-    ) {
+    /** @var IdentifierInterface[] */
+    private readonly array $items;
+
+    /**
+     * @param IdentifierInterface[] $items
+     */
+    public function __construct(array $items)
+    {
+        /** @var IdentifierInterface[] $values */
+        $values = array_values(array_filter($items, fn ($item) => $item instanceof MetadataInterface));
+        $keys = array_map(fn ($item) => $item->getName(), $values);
+        $this->items = array_combine($keys, $values);
     }
 
     public function getMetadata(): array
     {
-        return array_filter($this->items, fn ($item) => $item instanceof MetadataInterface);
+        return array_values($this->items);
     }
 
     public function get(PromptGet $input): PromptGetResult
     {
-        foreach ($this->items as $item) {
-            if ($item instanceof PromptGetterInterface && $input->name === $item->getName()) {
-                try {
-                    return $item->get($input);
-                } catch (\Throwable $e) {
-                    throw new PromptGetException($input, $e);
-                }
+        $item = $this->items[$input->name] ?? null;
+        if (!empty($item) && $item instanceof PromptGetterInterface) {
+            try {
+                return $item->get($input);
+            } catch (\Throwable $e) {
+                throw new PromptGetException($input, $e);
             }
         }
 

src/Capability/ResourceChain.php

@@ -16,28 +16,33 @@
  */
 class ResourceChain implements CollectionInterface, ResourceReaderInterface
 {
-    public function __construct(
-        /**
-         * @var IdentifierInterface[]
-         */
-        private readonly array $items,
-    ) {
+    /** @var IdentifierInterface[] */
+    private readonly array $items;
+
+    /**
+     * @param IdentifierInterface[] $items
+     */
+    public function __construct(array $items)
+    {
+        /** @var IdentifierInterface[] $values */
+        $values = array_values(array_filter($items, fn ($item) => $item instanceof MetadataInterface));
+        $keys = array_map(fn ($item) => $item->getUri(), $values);
+        $this->items = array_combine($keys, $values);
     }
 
     public function getMetadata(): array
     {
-        return array_filter($this->items, fn ($item) => $item instanceof MetadataInterface);
+        return array_values($this->items);
     }
 
     public function read(ResourceRead $input): ResourceReadResult
     {
-        foreach ($this->items as $item) {
-            if ($item instanceof ResourceReaderInterface && $input->uri === $item->getUri()) {
-                try {
-                    return $item->read($input);
-                } catch (\Throwable $e) {
-                    throw new ResourceReadException($input, $e);
-                }
+        $item = $this->items[$input->uri] ?? null;
+        if (!empty($item) && $item instanceof ResourceReaderInterface) {
+            try {
+                return $item->read($input);
+            } catch (\Throwable $e) {
+                throw new ResourceReadException($input, $e);
             }
         }
 

src/Capability/ToolChain.php

@@ -16,28 +16,33 @@
  */
 class ToolChain implements ToolExecutorInterface, CollectionInterface
 {
-    public function __construct(
-        /**
-         * @var IdentifierInterface[] $items
-         */
-        private readonly array $items,
-    ) {
+    /** @var IdentifierInterface[] */
+    private readonly array $items;
+
+    /**
+     * @param IdentifierInterface[] $items
+     */
+    public function __construct(array $items)
+    {
+        /** @var IdentifierInterface[] $values */
+        $values = array_values(array_filter($items, fn ($item) => $item instanceof MetadataInterface));
+        $keys = array_map(fn ($item) => $item->getName(), $values);
+        $this->items = array_combine($keys, $values);
     }
 
     public function getMetadata(): array
     {
-        return array_filter($this->items, fn ($item) => $item instanceof MetadataInterface);
+        return array_values($this->items);
     }
 
     public function call(ToolCall $input): ToolCallResult
     {
-        foreach ($this->items as $item) {
-            if ($item instanceof ToolExecutorInterface && $input->name === $item->getName()) {
-                try {
-                    return $item->call($input);
-                } catch (\Throwable $e) {
-                    throw new ToolExecutionException($input, $e);
-                }
+        $item = $this->items[$input->name] ?? null;
+        if (!empty($item) && $item instanceof ToolExecutorInterface) {
+            try {
+                return $item->call($input);
+            } catch (\Throwable $e) {
+                throw new ToolExecutionException($input, $e);
             }
         }